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

Revert "Attach transforms to model" #9036

Closed
wants to merge 1 commit into from

Conversation

glenn-jocher
Copy link
Member

@glenn-jocher glenn-jocher commented Aug 19, 2022

Reverts #9028

πŸ› οΈ PR Summary

Made with ❀️ by Ultralytics Actions

🌟 Summary

Optimizations and Updates in Classification Training and Validation.

πŸ“Š Key Changes

  • Attached class names directly to the model within the training script for better clarity and consistency.
  • Updated the code to work with FP16 (half-precision) inference, promoting better performance on compatible GPUs.
  • Changed the way predictions and images are processed and displayed, involving half-precision adjustments and clean-up in variable usage.

🎯 Purpose & Impact

  • πŸ” Streamlining the association of class names to improve the readability of the code and ensure class names are consistent throughout the training process.
  • ⚑ Enhancing performance by enabling half-precision inference. This can significantly accelerate model inference on GPUs with Tensor Cores, making it more efficient.
  • πŸ–Ό Improving image handling and prediction visualization during validation, ensuring that there's a correct display of images and predictions, aligned with the newly applied half-precision computations. This can lead to a smoother debugging and validation process for developers and clearer results for users.

@glenn-jocher glenn-jocher deleted the revert-9028-glenn-jocher-patch-1 branch August 19, 2022 14:53
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

Successfully merging this pull request may close these issues.

1 participant