MeioJane / CHR

SIXray : A Large-scale Security Inspection X-ray Benchmark in CVPR 2019
80 stars 37 forks source link

How Debug MulticlassTop5Engine? #23

Open Youngchen03 opened 1 year ago

Youngchen03 commented 1 year ago

Hello, I want to use MulticlassTop5Engine(Class) to measure accuracy. But get some errors:
"input_var = torch.autograd.Variable(self.state['input'])
TypeError: Variable data has to be a tensor, but got list" But when using MultiLabelMAPEngine(Class), not those errors.

Can you add the code of how to run class MluticlassTop5Engine? Thanks!