DATA_PATH=/datain/v-yinju/rqvae-zzx/data/instruments-emb-llama.npy
CKPT_DIR=your_ckpt_save_directory # E.g., /datain/v-yinju/rqvae-zzx/model
mkdir -p $CKPT_DIR
python -u main.py \
  --num_emb_list 256 256 256 256 \
  --sk_epsilons 0.0 0.0 0.0 0.003 \
  --lr 1e-3 \
  --device cuda:0 \
  --batch_size 1024 \
  --data_path $DATA_PATH \
  --ckpt_dir $CKPT_DIR

# Infer item index
# python generate_indices.py \
#  --ckpt_path your_rqvae_model_path \ E.g., /datain/v-yinju/rqvae-zzx/model/20241127/best_collision_model.pth
#  --data_path $DATA_PATH \
#  --save_path your_index_save_path \ E.g., /datain/v-yinju/rqvae-zzx/model/20241127/indices.json
#  --device cuda:0