tanluren

Hi , https://drive.google.com/open?id=1iOsNqQSNanTrbtNYQ5WNSZyxtXHZfjfZ it's trained from scratch and for 150000 batches only,mAP 60 on VOC, more training steps are required. Training with pretrained weights on classification sets would be a...

sub64比6快？看看能不能装apex

上传的cfg不对，已更新

估计环境问题，python numpy torch的版本不匹配，建议换个版本试试

还有想问下你的显卡显存多大，设的batchsize多少

Python 3.6.7 absl-py 0.8.0 albumentations 0.3.3 apex 0.1 asn1crypto 1.2.0 astor 0.8.0 backcall 0.1.0 certifi 2019.9.11 cffi 1.13.2 chardet 3.0.4 colorama 0.4.1 cryptography 2.7 cycler 0.10.0 Cython 0.29.14 decorator 4.4.1...

https://github.com/tanluren/yolov3-channel-and-layer-pruning/blob/master/train.py#L366

你好，可以贴一下完整的截图么，包括你使用的命令

OK，已沟通，自定义的cfg，题主已改代码匹配，棒

coming soon