Not able to train Enlighten GAN

[Bhuvan-21]

From the ImageEnhancement Directory
Command used:
!python train/enlightengan_train.py --dirA /content/drive/MyDrive/final_dataset/trainA --dirB /content/drive/MyDrive/final_dataset/trainB --weight_init_gain 0.02 --scale_size 320 --crop_size 320 --patchD_3 5 --self_attention --times_residual --patchD --vgg --patch_vgg --use_ragan --hybrid_loss
Output:
2021-02-14 09:33:58.659623: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.10.1
Traceback (most recent call last):
File "train/enlightengan_train.py", line 7, in
from utils.image_pool import ImagePool
ModuleNotFoundError: No module named 'utils'

I have noticed that utils is one directory above the enlightengan_train.py file.
I even tried training from inside the train folder, same error.

[wkhademi]

Have you updated your PYTHONPATH to include the ImageEnhancement directory? This seemed to fix any issues I had with that.

You can do that using:
export PYTHONPATH=$PYTHONPATH:/path/to/ImageEnhancement

where "/path/to/ImageEnhancement" is you actual path to this directory.

Also this only updates the current bash sessions PYTHONPATH environment variable so you will have to do this every time you open up a new terminal. You can add that line to your bashrc file to not have to do this every time you open a new terminal.