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

Tiny yolo v3 Training Error #8

Open
tmsor7 opened this issue Aug 26, 2020 · 2 comments
Open

Tiny yolo v3 Training Error #8

tmsor7 opened this issue Aug 26, 2020 · 2 comments

Comments

@tmsor7
Copy link

tmsor7 commented Aug 26, 2020

When I try to training this program using tiny yolo v3 (net_type=tiny), I got the under
messages.

train.py

Instructions for updating:
Use standard file APIs to check for files with this prefix.
0%| | 0/39782 [00:00<?, ?it/s]2020-08-26 16:00:13.648868: I tensorflow/stream_executor/platform/default/dso_loader.cc:42] Successfully opened dynamic library libcudnn.so.7
Traceback (most recent call last):
File "/usr/local/lib/python3.6/dist-packages/tensorflow/python/client/session.py", line 1356, in _do_call
return fn(*args)
File "/usr/local/lib/python3.6/dist-packages/tensorflow/python/client/session.py", line 1341, in _run_fn
options, feed_dict, fetch_list, target_list, run_metadata)
File "/usr/local/lib/python3.6/dist-packages/tensorflow/python/client/session.py", line 1429, in _call_tf_sessionrun
run_metadata)
tensorflow.python.framework.errors_impl.InvalidArgumentError: 2 root error(s) found.
(0) Invalid argument: Incompatible shapes: [2,18,18,3,4] vs. [2,36,36,3,4]
[[{{node define_loss/bigger_box_loss/logistic_loss_2/mul}}]]
[[define_loss/prob_loss/add/_69]]
(1) Invalid argument: Incompatible shapes: [2,18,18,3,4] vs. [2,36,36,3,4]
[[{{node define_loss/bigger_box_loss/logistic_loss_2/mul}}]]
0 successful operations.
0 derived errors ignored.

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
File "/media/sl/train.py", line 261, in
YoloTrain(net_type).train()
File "/media/sl/train.py", line 201, in train
self.trainable: True,})
File "/usr/local/lib/python3.6/dist-packages/tensorflow/python/client/session.py", line 950, in run
run_metadata_ptr)
File "/usr/local/lib/python3.6/dist-packages/tensorflow/python/client/session.py", line 1173, in _run
feed_dict_tensor, options, run_metadata)
File "/usr/local/lib/python3.6/dist-packages/tensorflow/python/client/session.py", line 1350, in _do_run
run_metadata)
File "/usr/local/lib/python3.6/dist-packages/tensorflow/python/client/session.py", line 1370, in _do_call
raise type(e)(node_def, op, message)
tensorflow.python.framework.errors_impl.InvalidArgumentError: 2 root error(s) found.
(0) Invalid argument: Incompatible shapes: [2,18,18,3,4] vs. [2,36,36,3,4]
[[node define_loss/bigger_box_loss/logistic_loss_2/mul (defined at media/sl/core/yolov3_tiny.py:197) ]]
[[define_loss/prob_loss/add/_69]]
(1) Invalid argument: Incompatible shapes: [2,18,18,3,4] vs. [2,36,36,3,4]
[[node define_loss/bigger_box_loss/logistic_loss_2/mul (defined at media/sl/core/yolov3_tiny.py:197) ]]
0 successful operations.
0 derived errors ignored.

Plz check my problems

@tmsor7 tmsor7 changed the title Tiny yolo v3 Build Error Tiny yolo v3 Training Error Aug 26, 2020
@tmsor7 tmsor7 closed this as completed Aug 27, 2020
@tmsor7 tmsor7 reopened this Aug 27, 2020
@highhand31
Copy link

I've got the same error when using "tiny"!!

@avBuffer
Copy link
Owner

Hey all, Thanks, could you mind telling which net type selecting on backbone, such as darknet53 or mobilenet?

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