python train.py --DATA=underwater_imagenet --EPOCHS=100 --NETWORK=pix2pix --L1_WEIGHT=100 --BATCH_SIZE=32 --IG_WEIGHT=0.0 --LEARNING_RATE=1e-4 --LOSS_METHOD=wgan
