virne.solver.learning.pg_cnn.pg_cnn_solver

Functions

make_policy(agent, **kwargs)
obs_as_tensor(obs, device)

Classes

PgCnnSolver(controller, recorder, counter, ...)

A Reinforcement Learning-based solver that uses Policy Gradient (PG) as the training algorithm and Convolutional Neural Network (CNN) as the neural network model.