somethingbyai commited on
Commit
4e28fb3
1 Parent(s): 073d382

Update helper.py

Browse files
Files changed (1) hide show
  1. helper.py +2 -1
helper.py CHANGED
@@ -111,7 +111,8 @@ def import_model_name_alt(model_x: nn.Module, activate_eval=True):
111
  def import_model_name_weights_copy(model_x: nn.Module, activate_eval=True):
112
  model_name = model_x._get_name()
113
  save_name_model = model_name + '_weights'
114
- PATH = '/Users/dominikocsofszki/PycharmProjects/mlp/data/weights_copy/' + save_name_model
 
115
  model_x.load_state_dict(torch.load(PATH))
116
  if activate_eval: model_x.eval()
117
  # print(f'{save_name_model} imported')
 
111
  def import_model_name_weights_copy(model_x: nn.Module, activate_eval=True):
112
  model_name = model_x._get_name()
113
  save_name_model = model_name + '_weights'
114
+ #PATH = '/Users/dominikocsofszki/PycharmProjects/mlp/data/weights_copy/' + save_name_model
115
+ PATH = save_name_model
116
  model_x.load_state_dict(torch.load(PATH))
117
  if activate_eval: model_x.eval()
118
  # print(f'{save_name_model} imported')