How to save and restore the parameters?#162

lesept777 wants to merge 3 commits into

lucidrains:mainlucidrains/vit-pytorch:mainfrom

kriventsov:mainkriventsov/vit-pytorch:mainCopy head branch name to clipboard