balbasty / torch-interpol

High-order spline interpolation in PyTorch
MIT License
62 stars 4 forks source link

OOM when called in a (long) loop #9

Closed balbasty closed 1 year ago

balbasty commented 1 year ago

It seems we have a similar issue as that described in these tickets:

There does not seem to be an official fix/workaround yet. I am opening this issue to keep track of it.

balbasty commented 1 year ago

It actually looks ok as long as all script functions are defined at the main level of a module (not nested into a function).

Nested definitions used to work and the nesting was treated as a namespace, but this is not the case anymore.