vit-pytorch icon indicating copy to clipboard operation
vit-pytorch copied to clipboard

Published 20 hours ago • verified publisher icon lucidrains

my own datasets? train.py pre.py?

Open henbucuoshanghai opened this issue 2 years ago • 2 comments

henbucuoshanghai avatar Apr 19 '22 08:04 henbucuoshanghai

You can look examples/cats_and_dogs.ipynb Good luck!

larsonwu0220 avatar May 17 '22 07:05 larsonwu0220

Hi, after training, how to save weights?

beizhengren avatar Jul 13 '23 09:07 beizhengren