在进行测试实验时,出现错误:
TypeError: ‘Baseline’ object is not iterable
这类问题主要是由于pytorch不同版本造成的。
解决方案:可以将
param_dict = torch.load(trained_path)
改为:
param_dict = torch.load(trained_path).state_dict()

在进行测试实验时,出现错误:
TypeError: ‘Baseline’ object is not iterable
这类问题主要是由于pytorch不同版本造成的。
解决方案:可以将
param_dict = torch.load(trained_path)
改为:
param_dict = torch.load(trained_path).state_dict()