Closed pawopawo closed 4 years ago
@pawopawo Thank you for your interest!
Your code is not update: https://github.com/TAMU-VITA/FasterSeg/blob/master/search/train_search.py#L66
Also please make sure the version of your thop
library matches the requirement.
thanks for your excellent work!
when I run
CUDA_VISIBLE_DEVICES=0 python train_search.py
, The code running here produces the above error