From acb9a0fec8d8a4dabeedcbb8e08c26f66d7083f0 Mon Sep 17 00:00:00 2001
From: haoneng.lhn <haoneng.lhn@alibaba-inc.com>
Date: 星期五, 08 十二月 2023 16:19:00 +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