You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
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:
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: