Behavioral Cloning

Actions space: Discrete | Continuous

Network Structure

../../../_images/pg.png

Algorithm Description

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.

  1. Sample a batch of transitions from the replay buffer.

  2. Use the current states as input to the network, and the expert actions as the targets of the network.

  3. For the network head, we use the policy head, which uses the cross entropy loss function.

class rl_coach.agents.bc_agent.BCAlgorithmParameters[source]