specify the meaning of logits in documentation (#238)
This commit is contained in:
parent
5ed6c1c7aa
commit
b364f1a26f
@ -92,6 +92,10 @@ It is also possible to use pre-defined MLP networks in :mod:`~tianshou.utils.net
|
||||
1. Input: observation ``obs`` (may be a ``numpy.ndarray``, ``torch.Tensor``, dict, or self-defined class), hidden state ``state`` (for RNN usage), and other information ``info`` provided by the environment.
|
||||
2. Output: some ``logits``, the next hidden state ``state``. The logits could be a tuple instead of a ``torch.Tensor``, or some other useful variables or results during the policy forwarding procedure. It depends on how the policy class process the network output. For example, in PPO :cite:`PPO`, the return of the network might be ``(mu, sigma), state`` for Gaussian policy.
|
||||
|
||||
.. note::
|
||||
|
||||
The logits here indicates the raw output of the network. In supervised learning, the raw output of prediction/classification model is called logits, and here we extend this definition to any raw output of the neural network.
|
||||
|
||||
|
||||
Setup Policy
|
||||
------------
|
||||
|
Loading…
x
Reference in New Issue
Block a user