AntixK / PyTorch-VAE

A Collection of Variational Autoencoders (VAE) in PyTorch.
Apache License 2.0
6.44k stars 1.05k forks source link

Mistake in experiment.py file #90

Open jarvis008 opened 5 months ago

jarvis008 commented 5 months ago

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!

MisterBourbaki commented 3 months ago

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