README.md

Training

python train.py --task gaussian2mnist --model tunet++ --filter_number 8

Testing

python train.py --task gaussian2mnist --model tunet++ --filter_number 8 --checkpoint /path/to/checkpooint.pth