...leak in GraphUNet due to function to_torch_csr_tensor...
Consecutive application of the forward-pass of GraphUNet causes a steady increase of memory, which eventually leads to a Memory Overflow. Changing the functionto_torch_csr_tensorinside the methodGraphUNet.augmen