TypeError: ‘Baseline‘ object is not iterable

关于复现行人重识别baseline

问题描述

在进行测试实验时，出现错误：
TypeError: ‘Baseline’ object is not iterable

原因分析：

这类问题主要是由于pytorch不同版本造成的。

解决方案：

可以将
param_dict = torch.load(trained_path)
改为：
param_dict = torch.load(trained_path).state_dict()