-
Notifications
You must be signed in to change notification settings - Fork 417
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
update to TensorFlow 2 #140
Comments
See https://github.com/ARM-software/ML-examples/tree/master/tflu-kws-cortex-m for tf2 version of training and quantizing. |
@Burton2000 Thanks, but how can I use the pre-trined models in the TensorFlow2 code? |
@Aishaj You can use the tflite files with the test_tflite.py file to evaluate the model accuracy on the test set or use the ckpt files with test.py, check the README in the Training folder for more details. You need to make sure the same arguments from training are used with test or test_tflite (see train_commands.txt) |
@Burton2000 Thanks, we would also appreciate adding label_wav.py file for TensorFlow 2 version. THank you |
Yep if that is something that will be of use we will look to add that as well when we have some time. |
Is there an update to work with tensorflow version 2, especially for train.py?
Indeed, the most problematic line is
from tensorflow.contrib import slim as slim
because there is no compatibility for this library between v1 and v2 of tf.I think other parts can be simply upgraded to version 2.
Thanks
The text was updated successfully, but these errors were encountered: