inferno-pytorch / inferno

A utility library around PyTorch
Other
244 stars 41 forks source link

Recursion error since there is no tensor/variable distinction #174

Closed sdamrich closed 5 years ago

sdamrich commented 5 years ago

Description

Initialising a model raised an recursion error. Specifically, in inferno/extensions/initializers/presets.py line 23

if isinstance(tensor, Variable):
    self.call_on_tensor(tensor.data)

the if clause is always true and one gets stuck in infinite recursion.

What I Did

Delete the lines

        if isinstance(tensor, Variable):
            self.call_on_tensor(tensor.data)
            return tensor