pytorch训练和验证

2022-01-20 15:59:45 阅读：136 来源： 互联网

标签：nn 训练验证 self torch pytorch print model 256

from audioop import bias
import torch
import torch.nn as nn

class SSD(nn.Module):
    def __init__(self):
        super(SSD, self).__init__()
        self.conv1 = nn.Conv2d(in_channels=3, out_channels=5, kernel_size=3, stride=2, padding=1, bias=False)

    def forward(self, x):
        x = self.conv1(x)
        return x

model = SSD()
model.train()
input = torch.ones([1,3,256,256])
output = model(input)
print(model.training)
print(output.shape)

print("*"*20)

with torch.no_grad():
    model.eval()
    input = torch.ones([1, 3, 256, 256])
    output = model(input)
    print(model.training)
    print(output.shape)

在model.eval()打开之后，model.training=False
在这里插入图片描述
在model内部的self.training=False，直接执行下一块result=self.postprocess(locs, confs)

标签：nn,训练,验证,self,torch,pytorch,print,model,256
来源： https://blog.csdn.net/jinxianwei1999/article/details/122603393

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9

pytorch训练和验证