Skip to content
Snippets Groups Projects
Commit a19f1f5c authored by Anjith GEORGE's avatar Anjith GEORGE
Browse files

Commented out model loading part

parent f2da377a
Branches
Tags
1 merge request!17Mccn extended
Pipeline #26549 failed
......@@ -93,20 +93,20 @@ class MCCNN(nn.Module):
print("light_cnn_model_file",light_cnn_model_file)
if not os.path.exists(light_cnn_model_file):
# if not os.path.exists(light_cnn_model_file):
bob.io.base.create_directories_safe(os.path.split(light_cnn_model_file)[0])
# bob.io.base.create_directories_safe(os.path.split(light_cnn_model_file)[0])
print('Downloading the LightCNN model')
# print('Downloading the LightCNN model')
bob.extension.download.download_file(url,light_cnn_model_file)
# bob.extension.download.download_file(url,light_cnn_model_file)
print('Downloaded LightCNN model to {}'.format(light_cnn_model_file))
# print('Downloaded LightCNN model to {}'.format(light_cnn_model_file))
## Loding the pretrained model for ch_0
# ## Loding the pretrained model for ch_0
self.load_state_dict(self.get_model_state_dict(light_cnn_model_file),strict=False)
# self.load_state_dict(self.get_model_state_dict(light_cnn_model_file),strict=False)
# copy over the weights to all other layers
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment