Pytorch Implementation Loss function Line 163 (main_train.py)

[dagarshali]

On line 163 we have (loss = criterion(model(batch_y), batch_x))

batch_y = clean + noise
model(batch_y) = x - output of the network.

If so, shouldn't the loss function be loss = criterion(model(batch_y),noise)?

[angry77cat]

I think the model is designed to predict noise