-
Notifications
You must be signed in to change notification settings - Fork 269
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
RuntimeError When Enabling Accuracy Checks in yolov3 Training on GPU. #2248
Labels
Comments
The following command will pass the test:
Looking into why the result is different in |
Fix: #2321 |
Thank you. I tested with your fix commit, and the accuracy test is passing. |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Issue Description
I encounter a RuntimeError related to gradient computation when enabling accuracy checks during the training of yolov3 in a GPU docker environment. The training runs without issues when the --accuracy flag is not used.
Steps to Reproduce
python install.py yolov3
python run.py yolov3 -d cuda -t train --accuracy
Expected Behavior
The training process should run without errors and perform accuracy checks without causing runtime errors.
Actual Behavior
The script executes successfully without the --accuracy flag.
However, when the accuracy check is enabled, it fails with the following error message:
TypeError: Darknet.forward() takes from 2 to 4 positional arguments but 6 were given
Running train method from yolov3 on cuda in eager mode with input batch size 4 and precision fp32.
env:pytorch-cuda=12.1 python=3.11
The text was updated successfully, but these errors were encountered: