AttributeError: 'DistributedDataParallel' object has no attribute 'module_list'

[shuncyu]

for idx in prune_idx:
bn_weights = gather_bn_weights(model.module_list, [idx])
tb_writer.add_histogram('before_train_perlayer_bn_weights/hist', bn_weights.numpy(), idx, bins='doane')

作者新加的这部分代码报错，如上

[tanluren]

我只有单卡，所以没弄多卡训练，你可以稍微改下，model.module.module_list，这样就不会报属性错误了。也可以参考这位朋友弄的多卡训练https://github.com/coldlarry/YOLOv3-complete-pruning/blob/master/train.py