Question: what is nn.module in pytorch?
Answer: PyTorch: Custom nn Modules A fully-connected ReLU network with one hidden layer, trained to predict y from x by minimizing squared Euclidean distance. This implementation defines the model as a custom Module subclass.