From 0eaeb071fe892a28b6cbfa7b11eef1b442a95e42 Mon Sep 17 00:00:00 2001
From: PascalIversen <p.iversen+1@live.de>
Date: Wed, 16 Oct 2024 13:15:22 +0200
Subject: [PATCH] fix nn

---
 drevalpy/models/simple_neural_network/utils.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/drevalpy/models/simple_neural_network/utils.py b/drevalpy/models/simple_neural_network/utils.py
index 3868462..da4d3e5 100644
--- 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