NilakshanKunananthaseelan

import torch.optim as optim

net = Net()
criterion = nn.MSELoss()
optimizer = optim.SGD(net.parameters(), lr=0.001, momentum=0.0,weight_decay=0.000)

NilakshanKunananthaseelan

import torch.nn as nn
import torch.nn.functional as F


class Net(nn.Module):
    def __init__(self):
        super().__init__()

        self.fc1 = nn.Linear(10, 100)
        self.fc2 = nn.Linear(100, 2)