attention-module icon indicating copy to clipboard operation
attention-module copied to clipboard

there still some error in the code

Open chenbohua3 opened this issue 7 years ago • 0 comments

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:)

chenbohua3 avatar Nov 09 '18 15:11 chenbohua3