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

Training steps automation #5236

Open
prayagpawar opened this issue Mar 11, 2024 · 1 comment
Open

Training steps automation #5236

prayagpawar opened this issue Mar 11, 2024 · 1 comment
Labels
enhancement Improvements or good new features

Comments

@prayagpawar
Copy link

Just like yolov8 is it possible for detectron2 to automate and export perfectly trained steps , As of now we are required to provide number of steps check for detection if not found train again , In yolov8 you just pass on the batch size and the training starts and it exports optimum model by checking validation loss .

@prayagpawar prayagpawar added the enhancement Improvements or good new features label Mar 11, 2024
@Programmer-RD-AI
Copy link
Contributor

Yes, it is possible to automate the training process in Detectron2 to some extent, similar to how YOLOv8 does it, but it requires setting up a custom training loop that can monitor the validation loss and determine the optimal number of training steps.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement Improvements or good new features
Projects
None yet
Development

No branches or pull requests

2 participants