### To train 

python train_neurips.py --config configs/neurips/choose_config 

### To test 
python test_neurips.py --config ....

### To create data 

python data/new_data/generate_dataset_final.py --config configs/data/data_confg 