export PYTHONPATH=./

attack_element=transition

env_name=halfcheetah
dataset=medium-replay
cuda_id=0
CUDA_VISIBLE_DEVICES=${cuda_id} python scripts/sequence_diffusion.py \
    --env_name ${env_name} \
    --dataset ${dataset} \
    --attack_element ${attack_element} \
    --dn_loops 1 \
    --dn_steps 80
    # --load_model_path models_cp/${env_name}_${dataset}