Comments (2)
Hello,
I am using tensorflow 2.3.1, but I was indeed setting Keras to channels_first in keras.json from previous experiments with CMSIS-NN (forgot about that one!). After fixing this (and applying the cudNN fix), both scripts ran with the expected results.
For information, with multiple GPUs I have to use this cudNN fix (loop over GPUs) or I get the error below:
for device in physical_devices:
tf.config.experimental.set_memory_growth(device, True)
ValueError: Memory growth cannot differ between GPU devices
Also for informaiton, the included weights.h failed at compile-time, from what I assume is an inconsistency with the recent interface changes. Here is the full compile log:
compile_log.txt
Thank you very much for the support, I will be moving on to trying it on my own model
from nnom.
Hi @Mjonir ,
May I know the TF version you are using? recommened TF2.2+. For the crash, are you setting Keras to 'CHW' or 'channel first'? The script currently only support the default HWC
shape format.
I am checking both attachement you provided.
Would you use the example weights.h in the example folder other that the one you converted from you model to test? see if the prediction is correct? for the mnist-simple example.
Thanks,
Jianjia
from nnom.
Related Issues (20)
- about reshape HOT 3
- 使用了per channel量化和kld量化方法后,出现了多次推理结果不一致的问题
- 关于输出维度的问题 HOT 4
- nnom静态内存支持如何打开? HOT 6
- main_pc.c中的test_x.txt和text_y.txt如何制作?
- 关于识别几秒时长的语音 HOT 3
- inhomogeneous shape after 1 dimensions
- keyword_spotting中的main_pc.c中的test_x.txt和text_y.txt如何制作?
- nothing
- 关于使用DW_Conv2D与Conv2D的移植后使用耗时的问题 HOT 1
- 使用#define NNOM_USING_CMSIS_NN 的错误。 HOT 2
- 运行rnnnoise的example报错 HOT 2
- Error: Deprecated Usage of `np.int` in `to_cstyle` Function in `gen_config.py`
- Keras 3 version compatibility HOT 1
- Installable using pip
- Exception with Tensorflow 2.16 HOT 2
- Consultation on kws examples?
- RNN state makes model output tend towards 0 HOT 2
- kws HOT 1
- RISCV rv32g undefined reference to `__divdi3'
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from nnom.