yolov5

yolov5 4.0版

AttributeError: Can't get attribute 'SiLU' on <module 'torch.nn.modules.activation'

网上的解决方法：

Update torch to 1.7

https://github.com/TrojanXu/yolov5-tensorrt

这个也有网络结构：

https://github.com/zgcr/pytorch-ImageNet-CIFAR-COCO-VOC-training/blob/c2f15210475a06415e6fc12e4cfcb5af938bb6f9/public/detection/models/yolov5.py

https://github.com/ultralytics/yolov5

测试代码：

1060上，batch size 12是ok的

  

   

    

     

    
    

     

      if __name__ == '__main__':
     
    
   

    

     

    
    

     

       parser = argparse.ArgumentParser()
     
    
   

    

     

    
    

     

       parser.add_argument('--cfg', typ
     
    
  
 

文章来源: blog.csdn.net，作者：网奇，版权归原作者所有，如需转载，请联系作者。

原文链接：blog.csdn.net/jacke121/article/details/106718075