Run it on Google Colab

Question

Run it on Google Colab

Patchwork53 opened this issue a year ago · 5 comments

Patchwork53 commented a year ago

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

Updated Colab Link

Answer 1 · 2023-11-07T22:23:50.000Z

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

Answer 2 · 2023-11-08T03:53:42.000Z

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

Answer 3 · 2023-11-08T04:26:07.000Z

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

Answer 4 · 2024-01-17T13:27:02.000Z

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?

Answer 5 · 2024-01-17T15:26:39.000Z

@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.