Error in the README

Assuming that you follow the exact instructions, and clone the repo recommended in the README (rather than this one), then the instruction

```
$ pip install tqdm  # install it if you haven't installed yes
$ python train.py
```
Should read

```
$ pip install tqdm  # install it if you haven't installed yes
$ python main.py
```