search

senmaoy / RAT-GAN

A conditional GAN for text-to-image
43 stars 9 forks source link

Where to change to apply resume_epoch functionality from saved NetG and NetD models? #18

Closed AudityGhosh closed 1 year ago

senmaoy commented 1 year ago

for evaluation: here for training: load the pre-trained models from here: https://github.com/senmaoy/RAT-GAN/blob/1e552d3f76fc9a176f21b4ec75b8568ab8ce5b8f/code/main.py#L109 using netG.load_state_dict(torch.load('../models/%s/netG_500.pth'%(cfg.CONFIG_NAME))) netD.load_state_dict(torch.load('../models/%s/netD_500.pth'%(cfg.CONFIG_NAME)))

AudityGhosh commented 1 year ago

Thank you very much, really grateful !