RuntimeError: volatile can only be set on leaf variables

output = model(*input)
output.volatile = False

for i in range(BATCH_SIZE):
    if nextState_batch[i][0][0].data[0] == 0:
        nextStateValues[i] = 0
        nextStateValues[i] = output[b[i]: b[i+1]].max(0)[0].data
expectedStateActionValues = (nextStateValues * GAMMA) +, -1)

loss = F.smooth_l1_loss(stateActionValues, Variable(expectedStateActionValues))

for param in model.parameters():
    if param.grad is not None:, 0.5)

This is the code snippet which is giving me the error, basically I am implementing a dqn and have followed the official dqn implementation given on the pytorch website.

Can someone please help resolve the runtime error.

Basically, you can only set variable.volatile = True on a variable that you just created. See this post for further explanation.
In short, you can not set volatile to True on a variable that is the result of an operation. If you want to set volatile to True, you can do it by calling output.detach_() if you don’t need to train model.

1 Like