search

amirbar / DETReg

Official implementation of the CVPR 2022 paper "DETReg: Unsupervised Pretraining with Region Priors for Object Detection".
https://amirbar.net/detreg
Apache License 2.0
334 stars 46 forks source link

checkpoint_args = torch.load(args.resume, map_location='cpu')['args'] KeyError: 'args'？？？ #68

Closed Caltech-Z closed 1 year ago

Caltech-Z commented 1 year ago

1: checkpoint_args = torch.load(args.resume,map_location='cpu')['args'] KeyError: 'args' I have this kind of error report in the evaluation stage, I don't know how to deal with it, I hope the owner can help me to solve it, thank you very much. 2: Does the test effect of a single GPU appear to be reduced?

amirbar commented 1 year ago

Couldn't reproduce it, if still relevant please describe the exact command you are running.