From ace42e5043be992c3a324fc637ba0594f10ad3b6 Mon Sep 17 00:00:00 2001
From: haoneng.lhn <haoneng.lhn@alibaba-inc.com>
Date: 星期五, 08 十二月 2023 16:39:38 +0800
Subject: [PATCH] fix loss normalization for ddp training
---
funasr/datasets/dataset.py | 6 ++++--
1 files changed, 4 insertions(+), 2 deletions(-)
diff --git a/funasr/datasets/dataset.py b/funasr/datasets/dataset.py
index 407f6aa..673a9b2 100644
--- a/funasr/datasets/dataset.py
+++ b/funasr/datasets/dataset.py
@@ -16,8 +16,10 @@
from typing import Mapping
from typing import Tuple
from typing import Union
-
-import h5py
+try:
+ import h5py
+except:
+ print("If you want use h5py dataset, please pip install h5py, and try it again")
import humanfriendly
import kaldiio
import numpy as np
--
Gitblit v1.9.1