moeen-m
/

whole

moeen-m commited on Aug 9, 2022

Commit

8e5b2fa

1 Parent(s): d9fab69

Update predefined.py

Files changed (1) hide show

predefined.py CHANGED Viewed

@@ -364,7 +364,7 @@ def evaluate(model, Ie,Be,Me,batch_size_e):
 bert_regressor = BertRegressor()
-bert_regressor.load_state_dict(torch.load("MAMBO_Cluster_deflection_training_impression_change",map_location=torch.device(device)))
 bert_regressor.eval()

 bert_regressor = BertRegressor()
+bert_regressor.load_state_dict(torch.load("MABMO_product",map_location=torch.device(device)))
 bert_regressor.eval()