https://github.com/ivan-zhou/cs236-project-conditional-gan
https://github.com/ivan-zhou/cs236-project-conditional-gan
Last synced: 3 months ago
JSON representation
- Host: GitHub
- URL: https://github.com/ivan-zhou/cs236-project-conditional-gan
- Owner: Ivan-Zhou
- Created: 2021-10-30T20:36:19.000Z (over 3 years ago)
- Default Branch: master
- Last Pushed: 2021-12-05T03:52:34.000Z (over 3 years ago)
- Last Synced: 2025-01-17T10:31:05.265Z (5 months ago)
- Language: Jupyter Notebook
- Size: 1.41 MB
- Stars: 0
- Watchers: 3
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
## Training
### Set up and launch docker
```
bash build.sh # build docker
bash launch.sh # launch docker
```
### Prepare Datasets
- `bitmoji-4k` dataset from Kaggle can be downloaded [here](https://drive.google.com/file/d/15XYVYe7wbc3yl1MaAMiwyFFpYvdH41g8/view?usp=sharing).
### Launch Training
To launch a training job, please pass two parameters:
- `DATASET`: The name of the Dataset for training (`stanford_dog` or `mnist`)
- `MODEL`: The type of the model for training (`default` or `cgan`)
For example:
```
# Train CGAN on MNIST
./run.sh mnist cgan
# Train CGAN on Stanford Dog
./run.sh stanford_dog cgan
./run.sh stanford_dogs_top_10 cgan
# Train CGAN on Fashion-MNIST
./run.sh fashion-mnist cgan
# Train CGAN on bitmoji-4k
./run.sh bitmoji-4k cgan
```
## Evaluation
To launch an eval job, please pass three parameters:
- `DATASET`: The name of the Dataset for evaluation (`stanford_dog` or `mnist`)
- `MODEL`: The type of the model for evaluation (`default` or `cgan`)
- `EXP_DIR`: The directory that contains checkpoints. The output sample png and metrics json will also be saved there
For example:
```
./run_eval.sh stanford_dog cgan default-project/out/cgan_stanford_dog
```