search

Lam1360 / YOLOv3-model-pruning

在 oxford hand 数据集上对 YOLOv3 做模型剪枝（network slimming）
MIT License
1.67k stars 432 forks source link

train error #69

Closed zhanghongruiupup closed 4 years ago

zhanghongruiupup commented 4 years ago

thank your code,it can work,but always show: /pytorch/aten/src/ATen/native/IndexingUtils.h:20: UserWarning: indexing with dtype torch.uint8 is now deprecated, please use a dtype torch.bool instead.

ignore this warning?