## Train conditional diffusion model

Change the envrionment name you want in config/env/atari.yaml and load reference policy in src/trainer.py

```
Python3 src/main.py
```

## Train Autoencoder-based detector
Load trajectory data in autoencoder_atari.py

```
Python3 autoencoder_atari.py
```

## Test Attack

```
Python3 src/test_ours.py --model "Defense Model" --env "Env_name" --attack_rate "attackfrequnecy" --attack "imagine(SHIFT-I)/real(SHIFT-O)"
```