Crate border_tch_agent

source · [−]
Expand description

RL agents implemented with tch.

Modules

cnn

Convolutional neural network.

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

TensorSubBatch

A buffer consisting of a Tensor.

Enums

Device

Device for using tch-rs.

Traits

ZeroTensor

Adds capability of constructing Tensor with a static method.