T5A_318的博客

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），会有BUG pytorch 踩坑之'tuple' object ha

########################################################################### 环境搭建 torch=1.6.0 cuda=10.1 python=3.7 1.mmcv安装 https://download.openmmlab.com/mmcv/dist/index.html 1）pip install mmcv (fail) 2) pip install mmcv_full *** (fail) 3)wg...