contrastive_unsupervised_rl

curl_learner

class xuance.torch.learners.contrastive_unsupervised_rl.curl_learner.CURL_Learner(config: Namespace, policy: torch.nn.Module, callback)

Bases: Learner

update(**samples)

class xuance.torch.learners.contrastive_unsupervised_rl.curl_learner.FrameStackTransform

Bases: object

drq_learner

Deep Q-Network (DQN) Paper link: https://www.nature.com/articles/nature14236 Implementation: Pytorch

class xuance.torch.learners.contrastive_unsupervised_rl.drq_learner.DrQ_Learner(config: Namespace, policy: torch.nn.Module)

Bases: Learner

update(**samples)

class xuance.torch.learners.contrastive_unsupervised_rl.drq_learner.FrameStackTransform

Bases: object

spr_learner

class xuance.torch.learners.contrastive_unsupervised_rl.spr_learner.FrameStackTransform

Bases: object

class xuance.torch.learners.contrastive_unsupervised_rl.spr_learner.SPR_Learner(config, policy, callback, temperature=0.1, tau=0.99, repr_lr=0.0001, prediction_steps=3)

Bases: Learner

update(**samples)