PyTorch-VAE icon indicating copy to clipboard operation
PyTorch-VAE copied to clipboard

Published 20 hours ago • verified publisher icon AntixK

Wrong import version [solved]

Open ZJLEOWANG3 opened this issue 2 years ago • 0 comments

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

ZJLEOWANG3 avatar Oct 21 '22 23:10 ZJLEOWANG3