search

NVlabs / NVAE

The Official PyTorch Implementation of "NVAE: A Deep Hierarchical Variational Autoencoder" (NeurIPS 2020 spotlight paper)
https://arxiv.org/abs/2007.03898
Other
999 stars 163 forks source link

Minor error in execution script for cifar-10 #3

Closed dylan-slack closed 3 years ago

dylan-slack commented 3 years ago

Thanks for this implementation!

I think the execution script for cifar-10 provided in the readme has an extra --save in it and should read:

export EXPR_ID=UNIQUE_EXPR_ID export DATA_DIR=PATH_TO_DATA_DIR export CHECKPOINT_DIR=PATH_TO_CHECKPOINT_DIR export CODE_DIR=PATH_TO_CODE_DIR cd $CODE_DIR python train.py --data $DATA_DIR/cifar10 --root $CHECKPOINT_DIR --save $EXPR_ID --dataset cifar10 \ --num_channels_enc 128 --num_channels_dec 128 --epochs 400 --num_postprocess_cells 2 --num_preprocess_cells 2 \ --num_latent_scales 1 --num_latent_per_group 20 --num_cell_per_cond_enc 2 --num_cell_per_cond_dec 2 \ --num_preprocess_blocks 1 --num_postprocess_blocks 1 --num_groups_per_scale 30 --batch_size 32 \ --weight_decay_norm 1e-2 --num_nf 1 --num_process_per_node 8 --use_se --res_dist --fast_adamax

It crashes as is right now.

arash-vahdat commented 3 years ago

Thanks for pointing this out. I just fixed it.