CogView icon indicating copy to clipboard operation
CogView copied to clipboard

After training, there are only 2G Pt files left. What's wrong with missing files?

Open tt-s-t opened this issue 3 years ago • 1 comments

After I tried to train with my own data set, the PT file I got was not 7g in size. Use it to execute. / scripts / text2image.sh -- debug error "runtimeerror: error (s) in loading state_dict for gpt2model:". Who can help me?Thank you.

tt-s-t avatar Aug 24 '22 01:08 tt-s-t

Is the model size identical to the size in the paper? You should match the hyperparameters.

Sleepychord avatar Aug 24 '22 01:08 Sleepychord