Found what appears to be a typo in train.py

[seungu-ji]

Looking through the code, I found a problem again.

In, train.py 153~156 lines.

model = D3().float()
model = model.to(device)
Loss_fun  = nn.MSELoss()
optimizer = optim.Adam(net.parameters(), lr = learning_rate)

optim.Adam receives net.parameters as a argument, and I think we should change that part to model.parameters or change the name of the model declared above to net.

How I changed it.

optimizer = optim.Adam(model.parameters(), lr = learning_rate)

or

net = D3().float()
net = net.to(device)