文章目錄
- 一、前言
- 二、我的代碼
- 三、我的錯誤
一、前言
在B站學習pytorch，看的是小土堆up主的視頻，歷經不少艱難險阻當學到“神經網路-卷積層這節內容時，代碼和視頻中的完全像同，但是我的卻出現了NotImplementedError的報錯，百思不得其解，枯了，希望能有人幫我看看
二、我的代碼
代碼如下（示例）：
import torch
from torch import nn
from torch.nn import Conv2d
from torch.utils.data import DataLoader
import torchvision
from torch.utils.tensorboard import SummaryWriter
dataset = torchvision.datasets.CIFAR10("../data",train=False,transform=torchvision.transforms.ToTensor(),download=True)
dataloader = DataLoader(dataset,batch_size=64)
class Tudui(nn.Module):
def __init__(self):
super(Tudui,self).__init__()
self.conv1 = Conv2d(in_channels=3,out_channels=6,kernel_size=3,stride=1,padding=0)
def foward(self,x):
x = self.conv1(x)
return x
tudui = Tudui()
#print(tudui)
writer = SummaryWriter("logs")
step = 0
for data in dataloader:
imgs,targets = data
output = tudui(imgs)
writer.add_image("input",imgs,step)
torch.reshape(output,(-1,3,30,30))
writer.add_image("output",output,step)
step = step+1
三、我的錯誤
希望有好心人撈撈我這個大一小白，萬分感謝
轉載請註明出處，本文鏈接：https://www.uj5u.com/qita/291413.html
標籤：AI
上一篇：吳恩達機器學習筆記1