pytorch学习3_长宽通道数-CSDN博客

本文链接：https://blog.csdn.net/Lhj0616/article/details/115758999

device = torch.device("cuda:0"if torch.cuda.is_available() else"cpu")
model.to(device)
#还有输入输出也放到显卡上
inputs,target = input.to(device),target.to(device)

当网络结构比较复杂，我们可以用新的类去封装他

你就像那个inception等等

$[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-qU1ltKbq-1618554035919)(C:\Users\haojie\AppData\Roaming\Typora\typora-user-images\image-20210415154137520.png)]$
虚线框这种将长，宽，通道数都进行了变化。而左边实线那种没有长宽通道数都没有发生变化
$[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-AjNtE6Fu-1618554035920)(C:\Users\haojie\AppData\Roaming\Typora\typora-user-images\image-20210415154321731.png)]$
虚线那个右边那个是进行了下采样，出现在conv3_1,conv4_1,conv5_1中
conv2_1不会出现。为什么呢，因为conv2_1(我们看50-layer,101-layer,152-layer)中最后输出的通道数是256(256是等于64乘4的)
18-layer,34-layer的不需要，不需要就可以满足，自然就不需要了

$[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-93vM1Cyy-1618554035921)(C:\Users\haojie\AppData\Roaming\Typora\typora-user-images\image-20210415155308964.png)]$