search

NVlabs / NVAE

The Official PyTorch Implementation of "NVAE: A Deep Hierarchical Variational Autoencoder" (NeurIPS 2020 spotlight paper)
https://arxiv.org/abs/2007.03898
Other
999 stars 163 forks source link

TypeError: batch_norm_backward_elemt() missing 1 required positional arguments: "count" #32

Open zhuofeng opened 2 years ago

zhuofeng commented 2 years ago

Seems a parameter is missing in the "torch.batch_norm_backward_elemt" function. Could anyone tell me how to solve this problem?

wangherr commented 2 years ago

I met the same error with pytorch 1.10 and python 3.10

while succeed with pytorch 1.8.0 and python 3.8.

Using the adviced version maybe work.

Hope this helps.