| 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 |