Expand description
RL agents implemented with tch.
Modules§
- cnn
- Convolutional neural network for Atari games.
- dqn
- DQN agent.
- iqn
- IQN agent.
- mlp
- Multilayer perceptron.
- model
- Interface of neural networks used in RL agents.
- opt
- Optimizers.
- sac
- SAC agent.
- util
- Utilities.
Structs§
- Tensor
Batch - A buffer consisting of a
Tensor.
Enums§
- Device
- Device for using tch-rs.
Traits§
- Zero
Tensor - Adds capability of constructing
Tensorwith a static method.