added possibility to skip last batch in training loop - #7

Merged

jkiesele merged 4 commits into

jkiesele:mainfrom

jonas-kleck:main

Jul 24, 2026

Commits on Jul 7, 2026

added possibility to skip last batch in training loop
Jonas Kleck
committed
9c7acf6

Commits on Jul 9, 2026

fix problem of double dropping
Jonas Kleck
committed
e61de7d
fix val_loop bug and dropping with drop_last_batch=False
Jonas Kleck
committed
fff04cb

Commits on Jul 24, 2026

added unittests for drop_last_batch functionality
Jonas Kleck
committed
5e47a8c