self.bias = nn.Parameter(torch.zeros(10))
def forward(self, xb):
return xb @ self.weights + self.bias
//////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
// Since we"re now using an object instead of just using a function, we
// first have to instantiate our model:
After Change
super().__init__()
self.lin = nn.Linear(784, 10)
def forward(self, xb):
return self.lin(xb)
//////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
// We instantiate our model and calculate the loss in the same way as before:
