Crate border_tch_agent

source ·
Expand description

RL agents implemented with tch.

Modules

  • Convolutional neural network.
  • DQN agent.
  • IQN agent.
  • Multilayer perceptron.
  • Interface of neural networks used in RL agents.
  • Optimizers.
  • SAC agent.
  • Utilities.

Structs

Enums

  • Device for using tch-rs.

Traits