ZJU-lishuang / yolov5_prune

yolov5 prune，Support V2, V3, V4 and V6 versions of yolov5

Apache License 2.0

559 stars 137 forks source link

WARNING: smart bias initialization failure. #46

Open hs7251912 opened 3 years ago

hs7251912 commented 3 years ago

在yolov5_prune/modelsori.py下测试DarkNet, if name =='main': img_size = 608 cfg = 'cfg/yolov5s_v4_hand.cfg' arc = 'default' device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

创建Darknet模型

model = Darknet(cfg, (img_size, img_size), arc=arc).to(device)

有警告： WARNING: smart bias initialization failure. WARNING: smart bias initialization failure. WARNING: smart bias initialization failure.

pytorch版本：1.8.1 python版本：3.7

ZJU-lishuang commented 3 years ago

代码使用torch版本为1.7，具体配置可参考yolov5第四版