[YOLOv5] Change default cache from ram to disk #257
Merged
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
For some quick benchmarking, I had two training_aware runs using the following model and data. One was with disk caching and one was with ram. For the first 50 epochs, the training time with disk was about 5 extra minutes (about an 8% slowdown) for disk caching vs ram.
Compared the following on disk caching vs ram for the first 50 epochs:
cv-detection
For the sake of time, I only looked at the time for the first 50 epochs but we could potentially consider a longer running job for additional benchmarking
To go back to ram, you can provide the the cache kwarg and set it to
ram