/mil

Code for "One-Shot Visual Imitation Learning via Meta-Learning"

Primary LanguagePythonMIT LicenseMIT

One-Shot Visual Imitation Learning via Meta-Learning

A TensorFlow implementation of the paper One-Shot Visual Imitation Learning via Meta-Learning (Finn*, Yu* et al., 2017). Here are the instructions to run our experiments shown in the paper.

First clone the fork of the gym repo found here, and following the instructions there to install gym. Switch to branch mil.

Then go to the mil directory and run ./scripts/get_data.sh to download the data.

After downloading the data, training and testing scripts for MIL are available in scripts/.

Note: The code only includes the simulated experiments.