Spaces:
Sleeping
Sleeping
somethingbyai
commited on
Commit
•
4e28fb3
1
Parent(s):
073d382
Update helper.py
Browse files
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')
|