Wrong import version [solved]

AntixK / PyTorch-VAE

A Collection of Variational Autoencoders (VAE) in PyTorch.

Apache License 2.0

6.44k stars 1.05k forks source link

Wrong import version [solved] #70

Open ZJLEOWANG3 opened 1 year ago

ZJLEOWANG3 commented 1 year ago

It seems like the torch version is too upgraded, so I try to change the requirement file as below and it works. pytorch-lightning>1.5.6 PyYAML==6.0 tensorboard==2.2.0 torch==1.12.1 torchsummary==1.5.1 torchvision>=0.10.1