search

ShuangXieIrene / ssds.pytorch

Repository for Single Shot MultiBox Detector and its variants, implemented with pytorch, python3.
MIT License
571 stars 166 forks source link

Which pytorch version ? #37

Open tzrtzr000 opened 5 years ago

tzrtzr000 commented 5 years ago

May I know which pytorch version you used for running this program? My current version is 0.4.1 but there are some variables initialization are being deprecated. (e.g. images = Variable(preproc(image, anno)[0].unsqueeze(0), volatile=True)).

Thanks

blueardour commented 5 years ago

It seems the author employ version before 0.4 when setting up the project. Many features after 0.4 are changed. such as the variable, the nogradient clause

blueardour commented 5 years ago

I did'nt meet the zero result, for example my test on mobilnet_v2_yolo_v2_voc gives a mAP of 0.721956. Check my result at https://github.com/ShuangXieIrene/ssds.pytorch/issues/41 I use pytorch 0.4.1