LeoWang311

icon company Cornell University icon location Ithaca NY icon location AI+Bio+Fin+Math :laughing:

Results 1 issues of LeoWang311

Wrong import version [solved]

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