# VGG

This repo is based on: https://github.com/pytorch/vision/blob/master/torchvision/models/vgg.py

## Inference on Single Image

```bash
bash infer.sh
```

#### Download Ofrecord

```bash
wget https://oneflow-public.oss-cn-beijing.aliyuncs.com/datasets/imagenette_ofrecord.tar.gz
tar zxf imagenette_ofrecord.tar.gz
```

### Run Oneflow Training script

```bash
bash train_oneflow.sh
```

