virne.solver.learning.pg_cnn2.solver

Functions

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

Classes

PgCnn2Solver(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.