|
004037947f
|
Fix reshape_classifier_output function to correctly reshape the final output layer (#13052)
* Fix reshape_classifier_output function to correctly reshape the final output layer
* Update torch_utils.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
---------
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: Ultralytics Assistant <135830346+UltralyticsAssistant@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
|
2024-05-29 22:12:17 +02:00 |
|