## Pre-training 

```
python main_pretrain.py \
    --batch_size 32 \
    --model etth_1024 \
    --epochs 400 \
    --root_path ${ROOT_DIR} \
    --data_path ${DATA_DIR} \
    --output_dir ${OUTPUT_DIR} \
    --log_dir ${LOG_DIR} \
    
```
## generative model training
Select the dataset in the generative_model.py file
```
# select 'sine'  'etth'  'energy'  'energy'
data = 'etth' 
```
run the python file
```
python generative_model.py
```
