PyTorch implementation of Episodic Meta Reinforcement Learning on variants of the "Two-Step" task. Reproduces the results found in three papers. Check the ReadMe for more details!
reinforcement-learning neuroscience lstm deepmind meta-learning a2c episodic-memory a2clstm differentiable-neural-dictionary meta-reinforcement-learning two-step-task
-
Updated
Dec 12, 2020 - Python