Error when trying to train mnist poincare experiment

emilemathieu / pvae

code for "Continuous Hierarchical Representations with Poincaré Variational Auto-Encoders".

https://arxiv.org/abs/1901.06033

MIT License

123 stars 43 forks source link

Error when trying to train mnist poincare experiment #8

Open saniazahan opened 3 years ago

saniazahan commented 3 years ago

I got the following error while testing the mnist poincare experiment. If I train for 79 epoch the error doesn't occur. Could you please suggest what is the problem

Ingmar-Bergman commented 3 months ago

Try to replace 'input = input.unsqueeze(-2).expand(input.shape[:-(len(input.shape) - 2)], self.out_features, self.in_features)' by 'input = input.unsqueeze(-2).expand(input.shape[:-1], self.out_features, self.in_features)'.

It worked for me.