yxgeee / MMT

[ICLR-2020] Mutual Mean-Teaching: Pseudo Label Refinery for Unsupervised Domain Adaptation on Person Re-identification.
https://yxgeee.github.io/projects/mmt
MIT License
472 stars 73 forks source link

an error in evaluator #37

Closed saga4me closed 3 years ago

saga4me commented 3 years ago

你好作者 在目录mmt的feature_extraction文件夹下的cnn.py文件,定义了一个extraction_cnn_feature函数 outputs=model(inputs) outputs包含了.data属性与.values属性 我在用其他backbone替换resnet时候会在evaluate时候报错 错误就是我的outputs没有这两个属性 请问作者 outputs.data 与outputs.values 这两个属性是在哪里定义的 我没有找到 辛苦作者回答一下 十分感谢

yxgeee commented 3 years ago

.data 就是pytorch的tensor放在cuda上的时候,就会自带这个属性。.values的话code里用不上,每次从14行就退出去了。