export WANDB_ENTITY=
export WANDB_API_KEY=
export WANDB_PROJECT=DiffusionNFT_PrefRestore
torchrun --nproc_per_node=8 --master_port=11234 scripts/train_nft_prefRestore.py --config config/pref_restore.py:pref_restore_multi_reward