Lightning-Universe/lightning-flash

Getting error while using backbones and head code

KVSAkhilesh opened this issue · 1 comments

KVSAkhilesh commented

Getting error while using backbones and head code
if below versions are not compatible please specify the versions we need to use to run the flash,
please mention what is the compatible torch version to use.

Code sample

from flash.image import ImageClassifier
backbones = ImageClassifier.available_backbones()
print(backbones

Environment

  • OS (e.g., Linux): Linux
  • Python version: 3.6.9

Error

image

Borda commented

seem you have some mismatch with versions, as if you would be using some of the last PL lightning-utilities are dependency of it or if you install latest Flash we have explicitly mentioned it too 🐰