解决RuntimeError: Error(s) in loading state_dict for Generator: Missing key(s) in state_dict:

出错截图
解决RuntimeError: Error(s) in loading state_dict for Generator: Missing key(s) in state_dict:_第1张图片
出错代码

generator.load_state_dict(checkpoint['generator'])

改正代码

generator.load_state_dict(checkpoint['generator'],strict=False)

运行成功

你可能感兴趣的:(大数据)