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

update to TensorFlow 2 #140

Open
upcoders opened this issue Jul 18, 2020 · 5 comments
Open

update to TensorFlow 2 #140

upcoders opened this issue Jul 18, 2020 · 5 comments

Comments

@upcoders
Copy link

upcoders commented Jul 18, 2020

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

@Burton2000
Copy link

See https://github.com/ARM-software/ML-examples/tree/master/tflu-kws-cortex-m for tf2 version of training and quantizing.

@Aishaj
Copy link

Aishaj commented Jul 22, 2021

@Burton2000 Thanks, but how can I use the pre-trined models in the TensorFlow2 code?

@Burton2000
Copy link

Burton2000 commented Jul 26, 2021

@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)

@Aishaj
Copy link

Aishaj commented Jul 27, 2021

@Burton2000 Thanks, we would also appreciate adding label_wav.py file for TensorFlow 2 version. THank you

@Burton2000
Copy link

Yep if that is something that will be of use we will look to add that as well when we have some time.

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

3 participants