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