Project-MONAI / Project-MONAI/MONAI
Cache data in both GPU memory and main memory
Open
@Nic-Ma is already working on this.
Since Dec 14, 2021.
enhancement
- Dominant language
- Python
- Stars
- 8.7k
- Forks
- 1.6k
- Avg merge
- 5d 1h
- Merged PRs (30d)
- 20
Description
Is your feature request related to a problem? Please describe.
This is interesting feedback from our internal SA team:
"Currently, the CacheDataset can obviously improve the training speed with caching either on main memory or on GPU memory. It would be nice to support caching on both sides to totally leverage the memory resources."
We can consider to set cache rate for both main memory and GPU memory to cache more data.
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Assessment
This issue has not been assessed yet.