File size: 663 Bytes
8a506a6 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 | 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 |