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
No related branches found
No related tags found
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