1

私は PyTorch を初めて使用し、画像分類に従来の MNIST データセットを使用しています。モデルのフィッティング中にエラーが発生しました:

NotImplementedError: uint8

fastai のライブラリのクラスを、すべてのトレーニング データと検証データ、および非常に基本的な単層ニューラル ネットワークのラッパーとして使用しています。私が使用しているコードは次のとおりです。

from keras.datasets import mnist

import matplotlib.pyplot as plt

from fastai.metrics import *

from fastai.model import *

from fastai.dataset import *

import torch.nn as nn

(x_train, y_train), (x_valid, y_valid) = mnist.load_data()

net = nn.Sequential(
  nn.Linear(784,10),
  nn.Softmax()).cuda()

md = ImageClassifierData.from_arrays('/data/mnist', 
(x_train,y_train), 
(x_valid, y_valid))   

loss = nn.NLLLoss()
metrics = [accuracy]
opt=optim.SGD(net.parameters(), 1e-1, momentum=0.9, weight_decay=1e-3)

fit(net, md, n_epochs=3, crit=loss, opt=opt, metrics=metrics)

誰かがこのエラーの内容と解決策を教えてもらえますか?

4

0 に答える 0