Should I put requires_grad=True
for input in below case:
>>> input = Variable(input_image_tensor, requires_grad=True)
>>> output = model(input)
>>> loss = loss_cal(input, output)
>>> loss.backward()
where the model
is encoder-decoder network.
Should I put requires_grad=True
for input in below case:
>>> input = Variable(input_image_tensor, requires_grad=True)
>>> output = model(input)
>>> loss = loss_cal(input, output)
>>> loss.backward()
where the model
is encoder-decoder network.