search

mpriessner / CAFI

Combined Repos from DAIN and Zooming SlowMo
MIT License
19 stars 5 forks source link

Running ZoomingSlowMo Colab tensorflow 2.x #4

Open Smooth-Cactus opened 1 year ago

Smooth-Cactus commented 1 year ago

Hello and first off, thank you for making your work publicly available !

I have been trying to run the ZoomingSlowMo Colab but Google Colab removed support for Tensorflow 1, and it is not possible to use %tensorflow_version 1.x magic anymore. I've tried to make it work with Tensorflow 2.x but keep getting during the requirement installation step. Is there by any chance an up-to-date version of the Colab available ? Thank you for your answer :)

mpriessner commented 7 months ago

Thank you!

Have a look at: https://github.com/mpriessner/CAFI/blob/main/ZoomingSlowMo_4_Microscopy_V2.ipynb

Here I removed the %tensorflow_version 1.x in the GPU checking step. The actual network uses pytorch for training and inference therefore the tensorflow version should not matter for running the actual notebook.