fix nn

daisybio · Oct 16, 2024 · 0eaeb07 · 0eaeb07
1 parent 78748fb
commit 0eaeb07
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/drevalpy/models/simple_neural_network/utils.py b/drevalpy/models/simple_neural_network/utils.py
@@ -155,6 +155,7 @@ def fit(
             shuffle=True,
             num_workers=num_workers,
             persistent_workers=True,
+            drop_last=True,
         )
 
         val_loader = None