rofunc.learning.RofuncRL.trainers.dtrans_trainer#

1.  Module Contents#

1.1.  Classes#

DTransTrainer

1.2.  Functions#

discount_cumsum

1.3.  API#

rofunc.learning.RofuncRL.trainers.dtrans_trainer.discount_cumsum(x, gamma)[source]#
class rofunc.learning.RofuncRL.trainers.dtrans_trainer.DTransTrainer(cfg, env, device, env_name, **kwargs)[source]#

Bases: rofunc.learning.RofuncRL.trainers.base_trainer.BaseTrainer

Initialization

load_dataset()[source]#

Load dataset from pickle file and preprocess it.

get_batch(batch_size=256)[source]#
train()[source]#

Main training loop.

post_interaction()[source]#