Models' Logits return,

[ajam74001]

Hi @takuseno, I was wondering how to return the logits (estimated Q values for each action) of CQL model (or any other Q based learners).

Cheers,
Many thanks for your help in advance,

[takuseno]

@ajam74001 Thanks for the issue. Unfortunately, there is no function by default. But, you can do it by accessing to internal objects:

# just example
# dqn needs to be built before this.
dqn = d3rlpy.algos.DQN()
dqn.impl.q_function(torch.rand((1, 10)))

[ajam74001]

Great! Many thanks for the response, dear @takuseno.