About the vision category
Topics related to either pytorch/vision or vision research related topics
smth
I created my model from scratch . I tried with batch_size 32 , it threw me this error. I tried with batch_size 1 , I got 320% accuracy (something wrong !). The error went away. so there is some dimensional mismatch  some where, I am lost! thanks a million !!! for your help and advise
bunnisiva
can we see your Model ?
JoBot_CoBot