Any plans to rewrite the code and remove tensorflow?

Deepshift / DeepCreamPy

deeppomf's DeepCreamPy + some updates

GNU Affero General Public License v3.0

424 stars 73 forks source link

Any plans to rewrite the code and remove tensorflow? #9

Open ConnoisseurProtege opened 3 weeks ago

ConnoisseurProtege commented 3 weeks ago

Seems that Tensorflow has been pretty much abandoned gpu support on Windows and Jax is used internally at Google, is there any plans to just switch to PyTorch? I am also unable to get my gpu to load the mosaic model even with the --is_mosaic=True argument.

naphteine commented 3 weeks ago

I do want to bring this project back to life, even under a new name and base if necessary, but I don't have the mental energy and time for it, unfortunately.

ConnoisseurProtege commented 3 weeks ago

Good thing is that the exe is basically limited by ram on modern systems, might be easy to implement a way to batch process. Right now I'm able to run 8 at a time with 64gb of ram(2 generation old processor doesn't even reach 50% usage)