what function module belongs to.

geyuying / PF-AFN

Official code for "Parser-Free Virtual Try-on via Distilling Appearance Flows", CVPR 2021.

545 stars 138 forks source link

Closed cloudy4next closed 3 years ago

cloudy4next commented 3 years ago

    save_checkpoint(model.module, os.path.join(opt.checkpoints_dir, opt.name, 'PBAFN_warp_epoch_%03d.pth' % (epoch+1)))

cloudy4next commented 3 years ago

Sorry for inconvenience

soumyajit123dev commented 3 years ago

@cloudy4next I am also getting a similar error. How did you solve it? error

I got the code to run if I remove the word "module". Basically, instead of "model.module", I just write "model". But is this the right way?

cloudy4next commented 3 years ago

Just remove (module) from each stage script 😃

module.model.something 
To
model.something

@soumyajit123dev