manpreet88 commited on
Commit
ebd0f34
·
1 Parent(s): 1de49d3

Update GINE.py

Browse files
Files changed (1) hide show
  1. 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 block preserved (same as original intent, using val_loader)
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 = [], []