Open DreamerrW opened 1 year ago
您好，我在运行你的代码时报了以下错误，您能解释一下可能报错的原因吗，感激不尽！
Backbone # param.: 23561205 Learnable # param.: 18043496 Total # param.: 41604701
Traceback (most recent call last): File "test.py", line 86, in model.load_state_dict(torch.load(args.load)['net']) # 这里有变化 KeyError: 'net'
您好，我在运行你的代码时报了以下错误，您能解释一下可能报错的原因吗，感激不尽！
Backbone # param.: 23561205 Learnable # param.: 18043496 Total # param.: 41604701
available GPUs: 2
Traceback (most recent call last): File "test.py", line 86, in
model.load_state_dict(torch.load(args.load)['net']) # 这里有变化
KeyError: 'net'