Actions space: Discrete|Continuous
Training the network
The replay buffer contains the expert demonstrations for the task. These demonstrations are given as state, action tuples, and with no reward. The training goal is to reduce the difference between the actions predicted by the network and the actions taken by the expert for each state.
- Sample a batch of transitions from the replay buffer.
- Use the current states as input to the network, and the expert actions as the targets of the network.
- The loss function for the network is MSE, and therefore we use the Q head to minimize this loss.