Spaces:
Running
Running
manpreet88 commited on
Commit ·
ebd0f34
1
Parent(s): 1de49d3
Update GINE.py
Browse files- PolyFusion/GINE.py +1 -1
PolyFusion/GINE.py
CHANGED
|
@@ -768,7 +768,7 @@ def train_and_evaluate(args: argparse.Namespace) -> None:
|
|
| 768 |
except Exception as e:
|
| 769 |
print(f"\nFailed to load best model from {best_model_path}: {e}")
|
| 770 |
|
| 771 |
-
# Final evaluation
|
| 772 |
model.eval()
|
| 773 |
preds_z_all, true_z_all = [], []
|
| 774 |
pred_dists_all, true_dists_all = [], []
|
|
|
|
| 768 |
except Exception as e:
|
| 769 |
print(f"\nFailed to load best model from {best_model_path}: {e}")
|
| 770 |
|
| 771 |
+
# Final evaluation
|
| 772 |
model.eval()
|
| 773 |
preds_z_all, true_z_all = [], []
|
| 774 |
pred_dists_all, true_dists_all = [], []
|