代码train.py中第125行代码中comput_loss应该为compute_loss，作者应该是误删了。 state = {'network': network.state_dict(), 'network_ema': ema.shadow, 'optimizer': optimizer.state_dict(), 'W': comput_loss.W,'epoch': args.start_epoch + epoch}
代码train.py中第125行代码中comput_loss应该为compute_loss，作者应该是误删了。
state = {'network': network.state_dict(), 'network_ema': ema.shadow, 'optimizer': optimizer.state_dict(), 'W': comput_loss.W,'epoch': args.start_epoch + epoch}