Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Upgrade to tf 2.8.2 #5

Open
wassimbensalem opened this issue Jan 17, 2024 · 0 comments
Open

Upgrade to tf 2.8.2 #5

wassimbensalem opened this issue Jan 17, 2024 · 0 comments

Comments

@wassimbensalem
Copy link

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()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant