-
I was able to get first scores for my trained segmentation network! However, when trying to parallelize, I noticed that my memory usage shoots up in htop (50G memory, 2048 dim, ~52k samples). I know the API says this step is memory intensive, but looking through the code, I have a hard time understanding why ---- we are just doing a Would someone have more insights here? |
Beta Was this translation helpful? Give feedback.
Replies: 1 comment
-
I see now it's the |
Beta Was this translation helpful? Give feedback.
I see now it's the
logger.debug
lines that floods the memory. updated the PR and removed those lines.