perceptual-reflection-removal icon indicating copy to clipboard operation
perceptual-reflection-removal copied to clipboard

Run it on Google Colab

Open Patchwork53 opened this issue 1 year ago • 5 comments

Made it work on Google Colab. Thanks, @ceciliavision for adding tf2 support, it was much easier this time.

Updated Colab Link

Patchwork53 avatar May 26 '23 15:05 Patchwork53

Hi @Patchwork53 , thanks so much! Although the link looks broken. Would you like to redeploy / re-enable it?

ceciliavision avatar Nov 07 '23 22:11 ceciliavision

Must have deleted it accidentally. Will redeploy it. The main.py file has a syntax error at line 20 btw.

Patchwork53 avatar Nov 08 '23 03:11 Patchwork53

The ckpt variable probably needs a NoneType check at line 354.

Patchwork53 avatar Nov 08 '23 04:11 Patchwork53

Hi @Patchwork53. Thanks for doing it on Google Colab. I'm trying to run, but both r_output and t_output results with the same results. Can you tell me why this might be the case?

Ekaterina245 avatar Jan 17 '24 13:01 Ekaterina245

@Ekaterina245 , how many epochs are you training?

Btw, my Colab code doesn't utilize the GPU. Tensorflow 2.10 doesn't even detect it (likely due to CUDA and cudnn version mismatch)... So training it pretty slow. Colab comes with tf 2.15 which does detect the GPU but that version isn't compatible with this repo. I couldn't find a way to install 2.10 with GPU support either.

Patchwork53 avatar Jan 17 '24 15:01 Patchwork53