JasonYUJX的博客

import collectionsimport mathimport randomimport sysimport timeimport osimport numpy as npimport torchfrom torch import nnimport torch.utils.data as Dataimport d2lzh_pytorch as d2lprint(t...

import mathimport torch import torch.nn as nnimport osdef file_name_walk(file_dir): for root, dirs, files in os.walk(file_dir):# print("root", root) # 当前目录路径 print("dirs"...

import numpy as npimport torchfrom torch import nn, optimimport torch.nn.functional as F载入数据import syssys.path.append("../input/")import d2l_jay9460 as d2ldevice = torch.device('cuda' if t...

#importimport sysimport d2lzh_pytorch as d2limport torchimport torch.nn as nnimport torch.optim as optimimport time#netclass Flatten(torch.nn.Module): #展平操作 def forward(self, x): ...

import timeimport mathimport numpy as npimport torchfrom torch import nn, optimimport torch.nn.functional as Fimport syssys.path.append("..") import d2lzh_pytorch as d2ldevice = torch.devic...

import torch import torchvisionimport numpy as npimport sysfrom torch import nnfrom torch.nn import init加载数据def load_data_fashion_mnist(batch_size, resize=None, root='~/Datasets/FashionMNI...