![](https://img-blog.csdnimg.cn/20201014180756925.png?x-oss-process=image/resize,m_fixed,h_224,w_224)
Pytorch
文章平均质量分 74
要好好学习Pytorch呀~
李丢丢要Up
一颗咸鸭蛋
展开
-
学习torchvision.models._dict_
torchvision.models._dict_[arch](pretrained=False)与python _dict_ 的关系 之前一直在写面向过程的C代码,转换到Python面向对象了有点不适应,前几天学习pytorch的时候,训练模型使用torchvision.models模块来加载预训练模型,记住了一种调用方法,如下:model = torchvison.models.__dict__[arch](pretrained=True) //arch是需要加载的预训练模型名,比如‘resnet原创 2021-05-26 13:30:33 · 5849 阅读 · 14 评论 -
Pytorch实现CenterNet—小白学习pytorch
Pytorch实现CenterNetPytorch实现CenterNetPycharm训练流程Pytorch实现CenterNet 第一次写技术相关的博客,想碎碎念一下:工作一年多以来始终用的是caffe,一直想切换成pytorch,但是没有项目跟进,只看了点中文教程,很难从理解怎样从理论到实际应用的过程。半个月前想了想:“种树的最好时间是十年前和现在”,所以决定手动踩坑。由于可怜打工人只有一块16XX的显卡,训练的batch_size只能放到2,中间拿模型进行测试,结果都是惨惨惨,没想到过了个周末原创 2021-05-24 12:02:48 · 929 阅读 · 2 评论