Suppose I don’t use torch.nn.Module and I have some trainable weight matrices w1 and w2. Then how do I feed these parameters into the optimiser? Would I do something like? :
torch.optim.SGD({w1,w2}, lr=1e-4)