getnamo / TensorFlow-Unreal

TensorFlow plugin for the Unreal Engine.
Other
1.15k stars 211 forks source link

Tensorflow 2 support #55

Open gokgozf opened 4 years ago

gokgozf commented 4 years ago

I am quite new to this plugin, this is more like a curiosity question. Will there be a tf2 support for the plugin, or it is only thought for 1x runtime

getnamo commented 4 years ago

In theory, modifying: https://github.com/getnamo/tensorflow-ue4/blob/master/Content/Scripts/upymodule.json to ```"tensorflow": "2.0.0" is all you should have to do. In practice there's either some mismatch between python environment or some module which doesn't grok well with the current embedded python. You may try to use a non-embedded version e.g.: https://github.com/getnamo/tensorflow-ue4/issues/42 for now which would allow you to setup your python environment however you want. There is also https://github.com/getnamo/machine-learning-remote-ue4 which will work with any backend that can run https://github.com/getnamo/ml-remote-server

I do hope to fix TF2 support in a future release (likely with an upgrade to the embedded python environment), but can't guarantee a timeline right now.