Reinforcement Learning, Gaming, Deep Learning.
configure_optimizers(self): return self.hparams.optim(self.policy.parameters(), lr=self.hparams.policy_lr) def train_dataloader(self): return DataLoader(dataset=self.dataset, batch_size=self.hparams.batch_size) # Training step. def training_step(self, batch, batch_idx): obs, action, returns = batch