Zibin Meng

to solve this problem, focus on line 181-182: ` torch.save(ema, os.path.join(opt.output_dir, now + 'ema.pth')) torch.save(ema2, os.path.join(opt.output_dir, now + 'ema2.pth'))` modify the code to the following: ` torch.save(ema.state_dict(), os.path.join(opt.output_dir, now +...