You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
For people who are trying to use this package with a new version of tensorflow. These are some of the changements that should be done in order to run this code using latest numpy and tf versions. If you have any other changements please feel free to add them.
Steps:
Clone scikit-tensor-py3 Git repo and delete all np.float statements. Replace them with float().
Install the repo again locally by pip install .
Replace
from tensorflow.python.keras.engine import InputSpec
by
from tensorflow.keras.layers import InputSpec
Replace trt usage
import tensorflow.contrib.tensorrt as trt
by
from tensorflow.python.compiler.tensorrt import trt_convert as trt
Delete all tf.Session and tf.reset_default_graph()
The text was updated successfully, but these errors were encountered:
For people who are trying to use this package with a new version of tensorflow. These are some of the changements that should be done in order to run this code using latest numpy and tf versions. If you have any other changements please feel free to add them.
Steps:
pip install .
from tensorflow.python.keras.engine import InputSpec
from tensorflow.keras.layers import InputSpec
import tensorflow.contrib.tensorrt as trt
from tensorflow.python.compiler.tensorrt import trt_convert as trt
The text was updated successfully, but these errors were encountered: