Hi @Azerus,
Thanks for your answer. Yeap, my guess was that the item() method is going to destroy the DAG. I got this from this article Memory Management, Optimisation and Debugging with PyTorch
I found my way around this problem by utilising GPU nodes with more memory, so that my pods are not evicted, but, of course for a higher price