Skip to content
Snippets Groups Projects
README.md 567 B
Newer Older
Dongmin Kim's avatar
Dongmin Kim committed
VQVAE2FC Test
Dongmin Kim's avatar
Dongmin Kim committed

* Install
  * Anaconda
  * conda env create -f environment.yml
    * (Includes python=3.6, torch=1.3, numpy=1.14)
  * [NVIDIA NCCL 2.4.8](https://developer.nvidia.com/nccl)
Dongmin Kim's avatar
Dongmin Kim committed
* Dataset
  * make dataset folder
  * make dataset/ffhq-dataset folder
  * Download [FFHQ dataset](https://github.com/NVlabs/ffhq-dataset)
Dongmin Kim's avatar
Dongmin Kim committed
* Run
Dongmin Kim's avatar
Dongmin Kim committed
  * Set visdom server IP (gins/global.gin)
  * Run visdom server
    * $> visdom
  * Set dataset, model (gins/global.gin)
  * Set compute nodes (gins/node_configs/default.gin)
  * Run script in every nodes
    * $> python main.py