
# README

## Configuration

See the [`environment.yaml`](./environment.yaml). We provide an user-friendly configuring method via Conda system, and you can create a new Conda environment using the command:

```
conda env create -f environment.yaml
```

## Training

 - Download datasets

```
sh ./prepare_before_train.sh
```

 - Train on CelebA-HQ

```
sh ./scripts/train_celeba_hq.sh
```

 - Train on AFHQ

```
sh ./scripts/train_afhq.sh
```