Closed ianpegg-bc closed 11 months ago
Thanks for catching this @ianpegg-bc. I'll have our engineering team look into this.
@ianpegg-bc Following up here. I've created a ticket to for the engineering team to address this. In the mean time, the work around would be move the tensor to CPU before logging it as a histogram, as you have suggested.
Describe the Bug
The application runs out of memory and is killed attempting to
log_histogram_3d
with a Pytorch tensor on the GPU.Expected behavior
Either of the following behaviors would be acceptable:
Where is the issue?
To Reproduce
The issue goes away when you set
device='cpu'
Stack Trace
stack trace if I stop it mid memory leak:
Link to Comet Project/Experiment
https://www.comet.ml/ianpegg-bc/test