epoch 100

JonathanEbner · Jan 11, 2024 · c21d00f · c21d00f
1 parent 5cd4a3c
commit c21d00f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/code/FM_NFM/main.py b/code/FM_NFM/main.py
@@ -217,8 +217,8 @@ def worker_init_fn(worker_id):
 
 
     if epoch % 10 == 0:
-        if epoch < 100:
-            continue
+        #if epoch < 100:
+            #continue
 
         model.eval()
         train_RMSE = evaluate.RMSE(model, args.model, train_loader)