| cd scripts | |
| CUDA_VISIBLE_DEVICES="0,1,2,3,4,5,6,7" torchrun --master_port 15000 --nproc_per_node=8 \ | |
| train_triplane_vae.py \ | |
| --configs ../configs/train_triplane_vae.yaml \ | |
| --accum_iter 2 \ | |
| --output_dir ../output/ae/chair \ | |
| --log_dir ../output/ae/chair --num_workers 8 \ | |
| --batch_size 22 \ | |
| --epochs 200 \ | |
| --warmup_epochs 5 \ | |
| --dist_eval \ | |
| --clip_grad 0.35 \ | |
| --category chair \ | |
| --data-pth ../data \ | |
| --replica 5 7 |