A bug for Resnet structure training

[May-forever]

Hi @erikguo and everyone

I find a bug when using this repo.

when I sparsity train resnet152_trident.cfg on my custom dataset. an error occur when I input the

following command:

python train.py --data data/obj.data --cfg cfg/resnet152_trident.cfg

The resnet152_trident.cfg can be find in darknet project: https://github.com/AlexeyAB/darknet/tree/master/cfg.

I find that all the Resnet-structure can not been sparsity train on this project.

Do you know what should I do now? I would be appreciated if you can give me a hand for fixing this bug.

Looking forward to hearing from you.

Thank you very much.

Best regards,
May