update packages

xunmengshe · Apr 27, 2023 · 6b9dde1 · 6b9dde1
1 parent 4dca53d
commit 6b9dde1
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 8 deletions.
diff --git a/py/g2p/requirements.txt b/py/g2p/requirements.txt
@@ -1,8 +1,6 @@
-editdistance==0.6.0
+editdistance==0.6.2
 hydra-core==1.3.2
-numpy==1.21.5
-omegaconf==2.1.1
-onnxruntime==1.10.0
-torch==1.10.2
-torchaudio==0.10.2
-tqdm==4.62.3
+omegaconf==2.1.2
+torch==2.0.0
+torchaudio==2.0.1
+tqdm==4.65.0
diff --git a/py/g2p/train.py b/py/g2p/train.py
@@ -49,7 +49,7 @@ def export(trainer, model_path, onnx_path):
         model=hydra.utils.instantiate(cfg),
         dataset=dataset,
         batch_size=256,
-        epochs=1)
+        epochs=10)
 
     train(trainer)