Spaces:

qgyd2021
/

call-audio-8

Running

HoneyTian commited on Jan 22

Commit

da6d9f1

1 Parent(s): 30c0bb4

update

Files changed (1) hide show

examples/vm_sound_classification/step_3_train_model.py CHANGED Viewed

@@ -197,7 +197,7 @@ def main():
         tgt_path = out_root / pretrained_model.stem
         model_pt_file = tgt_path / "model.pt"
-        with open(model_pt_file, "r") as f:
             state_dict = torch.load(f, map_location="cpu")
         model.load_state_dict(state_dict=state_dict)

         tgt_path = out_root / pretrained_model.stem
         model_pt_file = tgt_path / "model.pt"
+        with open(model_pt_file, "rb") as f:
             state_dict = torch.load(f, map_location="cpu")
         model.load_state_dict(state_dict=state_dict)