Reduce RAM usage: remove excess workers#118
Reduce RAM usage: remove excess workers#118nikich340 wants to merge 1 commit intojaywalnut310:mainfrom
Conversation
shivammehta25
left a comment
There was a problem hiding this comment.
It shouldn't be added to the official codebase, generally, people want to benefit from multiprocessing.
Have you looked into the commit? It changes hardcoded behaviour "I have 8 gpus, use 8 train loaders" to "I have N gpus, use N train loaders". |
|
@nikich340 num_workers has nothing to do with the number of GPUs you have. |
Ok, then there still should be a way to check if user really needs this, when every worker uses 4 gb of ram and it overfits real memory, using swap file give no benefit at all. |
Leaving 8 workers lead to x8 times more RAM usage when preloading mel spectrograms, when it is not needed for single-process training.