How to train my own dataset

[RamatovInomjon]

Hi @peterliht , Thanks for you great job!
I am trying to train my own dataset , however I got RuntimeError: size mismatch, m1: [2 x 2048], m2: [512 x 2] at errors,
I guess, it is because dataloader shape are different from yours , I am using this repo https://github.com/cs230-stanford/cs230-code-examples/tree/master/pytorch/vision to load my own data,
please guide to fix the issue
Thanks in advance, appreciate your time

[AEP-WYK]

Hi @peterliht , Thanks for you great job! I am trying to train my own dataset , however I got RuntimeError: size mismatch, m1: [2 x 2048], m2: [512 x 2] at errors, I guess, it is because dataloader shape are different from yours , I am using this repo https://github.com/cs230-stanford/cs230-code-examples/tree/master/pytorch/vision to load my own data, please guide to fix the issue Thanks in advance, appreciate your time

Hello, I ran into the same problem, can you tell me specifically how to modify it?