CSU-MS2-T2

Paused

Tingxie commited on Nov 22, 2024

Commit

ad2262b

1 Parent(s): effaac6

Update infer.py

Files changed (1) hide show

infer.py CHANGED Viewed

@@ -29,7 +29,6 @@ class ModelInference(object):
         self.model = ModelCLR(**self.config["model_config"]).to(self.device)
         state_dict = torch.load(pretrain_model_path,map_location=self.device)
         self.model.load_state_dict(state_dict)
-        self.model.to(device)
         self.model.eval()

         self.model = ModelCLR(**self.config["model_config"]).to(self.device)
         state_dict = torch.load(pretrain_model_path,map_location=self.device)
         self.model.load_state_dict(state_dict)
         self.model.eval()