a2c_networks¶
Module a2c_networks. Specifies various networks for A2C algorithm
- class a2c_networks.A2CNetSimpleLinear(n_columns: int, n_actions: int)¶
A2CNetSimpleLinear. Specifies a network architecture consisting of three linear layers
- __init__(n_columns: int, n_actions: int)¶
Constructor.
- Parameters
n_columns (Number of columns) –
n_actions (Number of actions) –
- forward(x: Tensor) tuple¶
Pass the state from the network
- Parameters
x (The torch tensor that represents the state) –
- Return type
The actor and the critic values