DingXiaoH / RepVGG

RepVGG: Making VGG-style ConvNets Great Again
MIT License
3.31k stars 434 forks source link

RepVGGplus-L2pse how to compute acc top 1 top 5 #72

Closed chgit0214 closed 3 years ago

chgit0214 commented 3 years ago

AttributeError: 'dict' object has no attribute 'topk'

DingXiaoH commented 3 years ago

Just use the 'main' item of the dict. Please check test.py (line 86).

chgit0214 commented 3 years ago

get it,thanks your reply