摘要:
好歹是把简单的图像分类弄了,用了谷歌的免费GPU ###1.加载GPU use_gpu = torch.cuda.is_available() device = torch.device('cuda' if torch.cuda.is_available() else 'cpu') print(de 阅读全文
posted @ 2022-05-15 18:45
姬桑day1
阅读(31)
评论(0)
推荐(0)
摘要:
###torch.optim在使用的前得先实例化 为了构建一个optimizer,需要先给它一个包含了需要优化的参数(必须是Variable对象,比如model.parameter()就是常见的这个对象), ####举例 opter = optim.SGD(medel.parameters(),lr 阅读全文
posted @ 2022-05-05 17:38
姬桑day1
阅读(49)
评论(0)
推荐(0)
摘要:
###nn.Conv2d就是pytorch中定义的的一个普通的类,可以很方便的实现二维的卷积,输入输出必须是四维张量,另外它有五个主要的参数, in_channels(输入的通道数) out_channels(输出的通道数) kernel_size(卷积核的大小) stride(卷积核移动的步长) 阅读全文
posted @ 2022-05-02 08:36
姬桑day1
阅读(263)
评论(0)
推荐(0)
浙公网安备 33010602011771号