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)#
class rofunc.learning.RofuncRL.trainers.dtrans_trainer.DTransTrainer(cfg, env, device, env_name, **kwargs)#

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

Initialization

load_dataset()#

Load dataset from pickle file and preprocess it.

get_batch(batch_size=256)#
train()#

Main training loop.

post_interaction()#