i was just testing which batch size would work, for now i see that batch size 32 works, but sometimes it does sometimes it doesn’t. will do a little more testing and come back here.
there is no training loop, i am just checking if i can atleast do a forward pass across my entire dataset.