跑代码
文章平均质量分 60
wtu_zwx
这个作者很懒,什么都没留下…
展开
专栏收录文章
- 默认排序
- 最新发布
- 最早发布
- 最多阅读
- 最少阅读
-
pytorch
1.Pytorch查看网络结构from torchsummary import summary device = torch.device("cuda" if torch.cuda.is_available() else "cpu") model = models.resnet18().to(device) summary(model, (3,256,256))注意:以上三种方法在处理输出为tuple时(如:LSTM),会有BUGpytorch 踩坑之'tuple' object ha原创 2021-05-14 21:32:54 · 448 阅读 · 0 评论 -
mmdector 使用dectors 训练自己的数据集
###########################################################################环境搭建torch=1.6.0cuda=10.1python=3.71.mmcv安装 https://download.openmmlab.com/mmcv/dist/index.html1)pip install mmcv (fail)2) pip install mmcv_full *** (fail)3)wg...原创 2021-01-01 14:13:57 · 792 阅读 · 0 评论
分享