which version of pytorch and torchvision is required?

[wonchul-kim]

💡 Your Question

I got this error: cannot import name 'vit_b_16' from 'torchvision.models'

for this: from data_gradients.managers.segmentation_manager import SegmentationAnalysisManager

Is there required version of torch/torchvision ?

Versions

No response

[ofrimasad]

Hi @wonchul-kim vit models were introduced in torchvision 0.12 (require torch > 1.11) Hope that helps :)