All algorithm are derived from nnabla_rl.algorithm.Algorithm.
Note
Algorithm will run on cpu by default (No matter what nnabla context is set in prior to the instantiation). If you want to run the algorithm on gpu, set the gpu_id through the algorithm's config. Note that the algorithm will override the nnabla context when the training starts.
nnabla_rl.algorithm.AlgorithmConfig
nnabla_rl.algorithm.Algorithm
nnabla_rl.algorithms.a2c.A2CConfig
nnabla_rl.algorithms.a2c.A2C
nnabla_rl.algorithms.bcq.BCQConfig
nnabla_rl.algorithms.bcq.BCQ
nnabla_rl.algorithms.bear.BEARConfig
nnabla_rl.algorithms.bear.BEAR
nnabla_rl.algorithms.categorical_dqn.CategoricalDQNConfig
nnabla_rl.algorithms.categorical_dqn.CategoricalDQN
nnabla_rl.algorithms.ddpg.DDPGConfig
nnabla_rl.algorithms.ddpg.DDPG
nnabla_rl.algorithms.dqn.DQNConfig
nnabla_rl.algorithms.dqn.DQN
nnabla_rl.algorithms.gail.GAILConfig
nnabla_rl.algorithms.gail.GAIL
nnabla_rl.algorithms.iqn.IQNConfig
nnabla_rl.algorithms.iqn.IQN
nnabla_rl.algorithms.munchausen_dqn.MunchausenDQNConfig
nnabla_rl.algorithms.munchausen_dqn.MunchausenDQN
nnabla_rl.algorithms.munchausen_iqn.MunchausenIQNConfig
nnabla_rl.algorithms.munchausen_iqn.MunchausenIQN
nnabla_rl.algorithms.ppo.PPOConfig
nnabla_rl.algorithms.ppo.PPO
nnabla_rl.algorithms.qrdqn.QRDQNConfig
nnabla_rl.algorithms.qrdqn.QRDQN
nnabla_rl.algorithms.reinforce.REINFORCEConfig
nnabla_rl.algorithms.reinforce.REINFORCE
nnabla_rl.algorithms.sac.SACConfig
nnabla_rl.algorithms.sac.SAC
nnabla_rl.algorithms.icml2018_sac.ICML2018SACConfig
nnabla_rl.algorithms.icml2018_sac.ICML2018SAC
nnabla_rl.algorithms.td3.TD3Config
nnabla_rl.algorithms.td3.TD3
nnabla_rl.algorithms.trpo.TRPOConfig
nnabla_rl.algorithms.trpo.TRPO
nnabla_rl.algorithms.icml2015_trpo.ICML2015TRPOConfig
nnabla_rl.algorithms.icml2015_trpo.ICML2015TRPO