最近才发现pytorch原来内部写好了一些网络,我们可以直接调用,并不需要从原论文处复制粘贴!
还是挺好的,免去了一些麻烦!
网址:https://github.com/pytorch/vision/tree/master/torchvision/models
pytorch可以直接调用的网络
最新推荐文章于 2023-12-10 14:00:36 发布
最近才发现pytorch原来内部写好了一些网络,我们可以直接调用,并不需要从原论文处复制粘贴!
还是挺好的,免去了一些麻烦!
网址:https://github.com/pytorch/vision/tree/master/torchvision/models