search

eto-ai / rikai

Parquet-based ML data format optimized for working with unstructured data
https://rikai.readthedocs.io/en/latest/
Apache License 2.0
137 stars 19 forks source link

torchvision >= 0.10.0 #561

Closed da-tubi closed 2 years ago

da-tubi commented 2 years ago

Fixes #549 , to make the follow python snippet work:

from torchvision.models.detection.ssd import ssd300_vgg16
eddyxu commented 2 years ago

lgtm. do we need to upgrade torch version tho?

da-tubi commented 2 years ago

I did not find a specific reason (bugs or api changes) to upgrade pytorch.