I dont know much about pytorch but does any one know what this line means
model = Net()
will create an object of the Net
class, which is most likely derived from nn.Module
and assign this object to the model
variable.
The model
variable can then be used to execute the forward pass by passing an input tensor to it (output = model(input)
), to load a state_dict
(model.load_state_dict(state_dict)
) etc.
So it cannot be used for training
It can be used for training as it’s just the model object.
I would recommend to take a look at the tutorials and check its usage.