蓝灯123

导航

2022年1月21日 #

一个全连接网络模型，激活函数是ReLU， 具有一个隐藏层且没有偏差， 经过训练可以使用欧几里得误差根据x来预测y。

摘要： from __future__ import print_function import torch import torch.nn as nn import torchvision.datasets as dsets import torchvision.transforms as transfo 阅读全文

posted @ 2022-01-21 15:32 蓝灯123 阅读(135) 评论(0) 推荐(0) 编辑

class myFirstNetwork(nn.Module):

摘要： import torchimport torch.nn as nnimport torchvision.datasets as dsetsimport torchvision.transforms as transformsfrom torch.autograd import Variablefro 阅读全文

posted @ 2022-01-21 15:00 蓝灯123 阅读(56) 评论(0) 推荐(0) 编辑

获取数据模板

摘要： import torch import torch.nn as nn import torchvision.datasets as dsets import torchvision.transforms as transforms from torch.autograd import Variabl 阅读全文

posted @ 2022-01-21 14:47 蓝灯123 阅读(32) 评论(0) 推荐(0) 编辑