Mistake in experiment.py file

[jarvis008]

In the forward function in experiment.py file, shouldn't it be:
return self.model.forward(input, **kwargs),
instead of
return self.model(input, **kwargs)

If not, please explain why as I'm unable to understand. Thanks!

[Baptiste Morisse]

Behind the curtain, the call to model is exactly model.forward :)