PytorchのmodelをClassを使わないコードに変換したい
以下のクラスを使ったコードをクラスを使わないコードにしたいのですが、
どうすればよいのでしょうか？
forwardの設定の仕方がわかりません。
from torch import nn
from torch.nn import functional
class Model(nn.Module):
def __init__(self):
super(Model,self).__init__()
self.fc1 = nn.Linear(10,100)
self.fc2 = nn.Linear(100,10)
def forward(self,x):
x = self.fc1(x)
x = functional.relu(x)
x = self.fc2(x)
return x
from torch import nn
model = nn.Sequential()
model.add_module('fc1', nn.Linear(10,100))
model.add_module('relu', nn.ReLU())
model.add_module('fc2', nn.Linear(100,10))
0 likes
