Closed bnawras closed 5 months ago
I can see a similar issue here: https://github.com/pytorch/pytorch/issues/62066
Could you cross check again and see if it is really related to AugmentationSequence
?
@kwcckw I have not reviewed the details of the implementation of AugmentationSequence
, but this is directly or indirectly related to this module. This is what will happen if we run the same code without it:
persistent_workers=True
paper_phase = [
agr.NoiseTexturize(
sigma_range=(2, 3),
turbulence_range=(2, 3),
p=1
),
agr.BrightnessTexturize(
texturize_range=(0.999, 0.9999),
deviation=0.01,
p=1
),
]
memory usage:
Thanks, i was able to reproduce the problem and fix it with this workaround from my end:
https://github.com/sparkfish/augraphy/pull/418
Could you reinstall it from the repo and try again?
fixed!
There seems to be a memory leak in
augraphy.AugmentationSequence
My environment
Dockerfile
Requirements
Сode to reproduce the leak
:fire: workaround: don't use
AugmentationSequence
orpersistent_workers=False