Exemple #1
0
 def test_load_model(self):
     load_model(self.config_test, self.new_csv_path, self.weight_path)
Exemple #2
0
 def test_load_model(self):
     model = load_model(self.config_test, self.new_csv_path,
                        self.weight_path)
     self.assertIsInstance(model, object)
Exemple #3
0
 def test_load_model(self):
     model = load_model(self.config_test, self.new_csv_path,
                        self.weight_path)
     self.assertIsInstance(model, object)
     convert_to_torch_script(model, "test.pt")