#!/bin/bash
#SBATCH --job-name=var_p3
#SBATCH --partition=long
#SBATCH --nodes=1
#SBATCH --mem=20gb
#SBATCH --cpus-per-task 12
#SBATCH --nodelist=g[007-009]
#SBATCH --gpus-per-node=2
#SBATCH --time=5-00:00:00
#SBATCH --output /projects/yuanai/projects/WassersteinVQ/VAR/slurm/FFHQ/var_result_p3.out
#SBATCH --error /projects/yuanai/projects/WassersteinVQ/VAR/slurm/FFHQ/var_error_p3.out

source ~/.bashrc
conda activate /home/fangxian/packages/anaconda/envs/share_VAR
CUDA_VISIBLE_DEVICES="0,1" python -m torch.distributed.launch --nproc_per_node=2 --master_port=35533 train_var.py --dataset_name=FFHQ --global_batch_size 64 --factor 16 --resolution 256 --codebook_size 100000 --beta=0.1 --gamma=0.5