## Requirements

TensorFlow >= v1.9.0.

## Training

To replicate the results of the paper on a particular dataset, execute (see the Note below for downloading the CUB and AWA datasets):
```bash
$ ./replicate_results.sh <DATASET> <THREAD-ID> 
```

Example runs are:
```bash
$ ./replicate_results.sh MNIST 4     /* Train MEGA-2 on MNIST */

$ ./replicate_results.sh CIFAR 3     /* Train MEGA-2 on CIFAR */

$ ./replicate_results.sh CUB 3    /* Train MEGA-2 on CUB */

$ ./replicate_results.sh AWA 7     /* Train MEGA-2 on AWA */
```

### Note
For CUB and AWA experiments, download the dataset prior to running the above script. Run following for downloading the datasets:

```bash
$ ./download_cub_awa.sh
```
The plotting code is provided under the folder `plotting_code/`. Update the paths in the plotting code accordingly.
