osanseviero
/

hubert_s3prl_req

Automatic Speech Recognition

Model card Files Files and versions

Omar Sanseviero commited on Jul 15, 2021

Commit

23e43c0

·

1 Parent(s): b18f6c8

Fix char.dict finding

Files changed (1) hide show

model.py +1 -1

model.py CHANGED Viewed

@@ -19,7 +19,7 @@ class PreTrainedModel(Runner):
         ckp["Args"].init_ckpt = ckp_file
         ckp["Args"].mode = "inference"
         ckp["Args"].device = "cpu" # Just to try in my computer
-        ckp["Config"]["downstream_expert"]["datarc"]["dict_path"]='char.dict'
         Runner.__init__(self, ckp["Args"], ckp["Config"])

         ckp["Args"].init_ckpt = ckp_file
         ckp["Args"].mode = "inference"
         ckp["Args"].device = "cpu" # Just to try in my computer
+        ckp["Config"]["downstream_expert"]["datarc"]["dict_path"]=os.path.join(path,'char.dict')
         Runner.__init__(self, ckp["Args"], ckp["Config"])