nas_stegan

Datasets

CIFAR10

ImageNet can find here.

Experiments

You need to add your file path in the code first. Then run the following commands.

### for searching architecture
python main_nas.py

You can use our search structure to train directly or add your own search structure to genotypes.py to train.

### for training 
python main.py --arch 'your architecture'

You can modify the test_ckpt to test specified model.

### for testing
python main.py --mode test --test_ckpt 'your file path'

Pre-trained weights

Coming soon.

Acknowledgments

Codes are heavily borrowed from UDH and PC-DARTS.