Open feixiaoyu2223 opened 5 years ago
先是在运行sh make.sh后有如下warning: /home/wr506/Documents/flyfly/MOTDT/models/psroi_pooling/src/psroi_pooling_cuda.c:23:6: warning: unused variable ‘batch_size’ [-Wunused-variable] int batch_size = THCudaTensor_size(state, features, 0);
然后运行eval_mot.py后显示没有使用batchnorm层,有warning如下: loading layers from squeezenet1_1... 2019-04-02 14:55:27 [WARNING]: No layer: upconv_1.1.num_batches_tracked 2019-04-02 14:55:27 [WARNING]: No layer: proj_1.2.num_batches_tracked 2019-04-02 14:55:27 [WARNING]: No layer: upconv_2.1.num_batches_tracked 2019-04-02 14:55:27 [WARNING]: No layer: proj_2.2.num_batches_tracked 2019-04-02 14:55:27 [WARNING]: No layer: cls_conv.1.num_batches_tracked load cls model from: data/squeezenet_small40_coco_mot16_ckpt_10.h5
我用的是pyrorch0.4.1+python3.6版本,急求解答方案,谢谢。
num_batches_tracked 是pytorch后来加入的参数,不影响实际的计算。
num_batches_tracked
真是太感谢作者的回复了!
你好 你跑通了吗? 我的错误和你一样 我想用笔记本的cpu做实验,把程序中的cuda全注释换成了cpu ,也有和你一样的问题
先是在运行sh make.sh后有如下warning: /home/wr506/Documents/flyfly/MOTDT/models/psroi_pooling/src/psroi_pooling_cuda.c:23:6: warning: unused variable ‘batch_size’ [-Wunused-variable] int batch_size = THCudaTensor_size(state, features, 0);
然后运行eval_mot.py后显示没有使用batchnorm层,有warning如下: loading layers from squeezenet1_1... 2019-04-02 14:55:27 [WARNING]: No layer: upconv_1.1.num_batches_tracked 2019-04-02 14:55:27 [WARNING]: No layer: proj_1.2.num_batches_tracked 2019-04-02 14:55:27 [WARNING]: No layer: upconv_2.1.num_batches_tracked 2019-04-02 14:55:27 [WARNING]: No layer: proj_2.2.num_batches_tracked 2019-04-02 14:55:27 [WARNING]: No layer: cls_conv.1.num_batches_tracked load cls model from: data/squeezenet_small40_coco_mot16_ckpt_10.h5
我用的是pyrorch0.4.1+python3.6版本,急求解答方案,谢谢。