Skip to content
Snippets Groups Projects
Commit 519bd284 authored by Ryan Kiros's avatar Ryan Kiros
Browse files

Merge pull request #2 from ivendrov/load

added ability to specify custom model location without modifying the file
parents fa0ad405 430b3dc1
No related branches found
No related tags found
No related merge requests found
...@@ -17,10 +17,10 @@ from model import init_params, build_sentence_encoder, build_image_encoder ...@@ -17,10 +17,10 @@ from model import init_params, build_sentence_encoder, build_image_encoder
#-----------------------------------------------------------------------------# #-----------------------------------------------------------------------------#
# Specify model location here # Specify model location here
#-----------------------------------------------------------------------------# #-----------------------------------------------------------------------------#
path_to_model = '/ais/gobi3/u/rkiros/uvsmodels/coco.npz' default_model = '/ais/gobi3/u/rkiros/uvsmodels/coco.npz'
#-----------------------------------------------------------------------------# #-----------------------------------------------------------------------------#
def load_model(): def load_model(path_to_model=default_model):
""" """
Load all model components Load all model components
""" """
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment