/imagination_augmented_agents

Replicating Imagination-Augmented Agents for Deep Reinforcement Learning

Primary LanguagePython

imagination_augmented_agents

Replicating Imagination-Augmented Agents for Deep Reinforcement Learning

Pytorch code for training Imagination Augment Agents Paper using A2C/PPO.