model.onnx file is not creating
Hello,
I ran your code and it worked and created a model for me, but only in the h5 format, the model is not created in the onnx format. Apparently, I have a weak video card, it does not support calculations... This is what happened at the end of training the model: 2025-01-31 18:26:02.160816: F tensorflow/stream_executor/cuda/cuda_driver.cc:345] Check failed: CUDA_SUCCESS == cuDevicePrimaryCtxGetState(device, &former_primary_context_flags, &former_primary_context_is_active) (0 vs. 303)
But when I tried to check the model, I ran the file, an error occurred:
(venv) G:\testPyton>python inferenceModel.py
Traceback (most recent call last):
File "G:\testPyton\inferenceModel.py", line 31, in
I'm experiencing the same issue. Can anyone who knows help? 🙂
I still haven’t been able to overcome the problem, now I’m trying to rewrite the script from Python to node.js, I understand it more...