attention-module
attention-module copied to clipboard
there still some error in the code
Hi, thank you for your codes, it dose help me a lot:)
But there is still some error in the code like
transforms.RandomSizedCrop(size0) in train_imagenet.py
and it seems that you construct your optimizer before you move model parameter to gpus.
Could you please fix this error, thanks for your patience:)