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):
):
self.arch=arch
_model= FaceXZooModelFactory(self.arch)
# self.model = _model.get_model()
filename = _model.get_facexzoo_file()
checkpoint_name = _model.get_checkpoint_name()
config=None
......@@ -317,11 +314,6 @@ class FaceXZooModel(PyTorchModel):
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()
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