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

Custom model save_model issue #94

Open
AndrewMarines opened this issue Jul 26, 2022 · 2 comments
Open

Custom model save_model issue #94

AndrewMarines opened this issue Jul 26, 2022 · 2 comments

Comments

@AndrewMarines
Copy link

Hello, I downloaded an already done weights file for my purpose.
python save_model.py --weights path_to_custom_weights --output ./checkpoints/custom-416 --input_size 416 --model yolov4

but he's giving me this issue:
2022-07-26 10:55:32.916596: W tensorflow/stream_executor/platform/default/dso_loader.cc:64] Could not load dynamic library 'cudart64_110.dll'; dlerror: cudart64_110.dll not found 2022-07-26 10:55:32.916796: I tensorflow/stream_executor/cuda/cudart_stub.cc:29] Ignore above cudart dlerror if you do not have a GPU set up on your machine. 2022-07-26 10:55:36.224921: W tensorflow/stream_executor/platform/default/dso_loader.cc:64] Could not load dynamic library 'nvcuda.dll'; dlerror: nvcuda.dll not found 2022-07-26 10:55:36.225092: W tensorflow/stream_executor/cuda/cuda_driver.cc:269] failed call to cuInit: UNKNOWN ERROR (303) 2022-07-26 10:55:36.229386: I tensorflow/stream_executor/cuda/cuda_diagnostics.cc:169] retrieving CUDA diagnostic information for host: AE 2022-07-26 10:55:36.229590: I tensorflow/stream_executor/cuda/cuda_diagnostics.cc:176] hostname: AE 2022-07-26 10:55:36.230180: I tensorflow/core/platform/cpu_feature_guard.cc:193] This TensorFlow binary is optimized with oneAPI Deep Neural Network Library (oneDNN) to use the following CPU instructions in performance-critical operations: AVX AVX2 To enable them in other operations, rebuild TensorFlow with the appropriate compiler flags. Traceback (most recent call last): File "C:\Users\andre\Desktop\yolov4-custom-functions-master\save_model.py", line 58, in <module> app.run(main) File "C:\Users\andre\AppData\Local\Programs\Python\Python310\lib\site-packages\absl\app.py", line 308, in run _run_main(main, args) File "C:\Users\andre\AppData\Local\Programs\Python\Python310\lib\site-packages\absl\app.py", line 254, in _run_main sys.exit(main(argv)) File "C:\Users\andre\Desktop\yolov4-custom-functions-master\save_model.py", line 54, in main save_tf() File "C:\Users\andre\Desktop\yolov4-custom-functions-master\save_model.py", line 49, in save_tf utils.load_weights(model, FLAGS.weights, FLAGS.model, FLAGS.tiny) File "C:\Users\andre\Desktop\yolov4-custom-functions-master\core\utils.py", line 143, in load_weights conv_weights = conv_weights.reshape(conv_shape).transpose([2, 3, 1, 0]) ValueError: cannot reshape array of size 4554552 into shape (1024,512,3,3)

@Bayati86
Copy link

Bayati86 commented Dec 7, 2022

same problem did you solve it?

@AndrewMarines
Copy link
Author

Nope and i don't remember anymore the code i was trying to create.

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

2 participants