I am trying to implement the "SeGAN: Segmenting and Generating the invisible" paper on ubuntu 18.04 with Geforce RTX 2060. I have installed the Driver, CUDA, cuDNN