python 3.7.9

pytorch 1.6.0

tensorboard 2.3.0 

torchvision 0.7.0 


train: 

```python
python main.py 
```

eval: 

```python
python main.py --train 0 
```

```python
tensorboard --logdir logs 
```