search

Gumpest / YOLOv5-Multibackbone-Compression

YOLOv5 Series Multi-backbone(TPH-YOLOv5, Ghostnet, ShuffleNetv2, Mobilenetv3Small, EfficientNetLite, PP-LCNet, SwinTransformer YOLO), Module(CBAM, DCN), Pruning (EagleEye, Network Slimming), Quantization (MQBench) and Deployment (TensorRT, ncnn) Compression Tool Box.
994 stars 200 forks source link

训练时候loss都是nan，是什么原因？是预训练权重的原因吗 #98

Closed gitshitou closed 6 months ago

gitshitou commented 7 months ago

训练时候loss都是nan，是什么原因？是预训练权重的原因吗

Gumpest commented 6 months ago

你好 @gitshitou，应该是模型的Bug，先调小loss weight试试