python main.py --dataset=underwater_imagenet --continue_train=1 --checkpoint_dir=checkpoint/underwater_imagenet_256/
