#!/bin/bash
#SBATCH --job-name=vqwae_p1
#SBATCH --partition=gpuv100
#SBATCH --nodes=1
#SBATCH --mem=20gb
#SBATCH --cpus-per-task 8
#SBATCH --gres=gpu:1
#SBATCH --time=1-00:00:00
#SBATCH --output /mmfs1/data/fangxian/ICML2025/slurm/CIFAR10/vqwae_result_p1.out
#SBATCH --error /mmfs1/data/fangxian/ICML2025/slurm/CIFAR10/vqwae_error_p1.out

conda activate share_VAR 
CUDA_VISIBLE_DEVICES="0" python train_vqvae.py --quantizer_name=vqwae --dataset_name=CIFAR10 --batch_size 128 --factor 4 --codebook_size 512 --beta=1.0 --gamma=0.0