Commit 06a218e6 authored by Anjith GEORGE's avatar Anjith GEORGE
Browse files

Cleanup

parent 3b73890d
Pipeline #52784 failed with stage
in 48 minutes and 46 seconds
...@@ -292,10 +292,7 @@ class FaceXZooModel(PyTorchModel): ...@@ -292,10 +292,7 @@ class FaceXZooModel(PyTorchModel):
): ):
self.arch=arch self.arch=arch
_model= FaceXZooModelFactory(self.arch) _model= FaceXZooModelFactory(self.arch)
# self.model = _model.get_model()
filename = _model.get_facexzoo_file() filename = _model.get_facexzoo_file()
checkpoint_name = _model.get_checkpoint_name() checkpoint_name = _model.get_checkpoint_name()
config=None config=None
...@@ -317,11 +314,6 @@ class FaceXZooModel(PyTorchModel): ...@@ -317,11 +314,6 @@ class FaceXZooModel(PyTorchModel):
self.model = _model.get_model() self.model = _model.get_model()
print('self.checkpoint_path',self.checkpoint_path)
print('self.arch',self.arch)
#print('self.model', self.model)
model_dict = self.model.state_dict() model_dict = self.model.state_dict()
pretrained_dict = torch.load(self.checkpoint_path, map_location=torch.device('cpu'))['state_dict'] pretrained_dict = torch.load(self.checkpoint_path, map_location=torch.device('cpu'))['state_dict']
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment