- Project needs to use [torchvision 0.11.0](https://pytorch.org/vision/0.11/models.html) where it provides follwowing pretrained models. ```python import torchvision.models as models mobilenet_v3_large = models.mobilenet_v3_large() mobilenet_v3_small = models.mobilenet_v3_small() efficientnet_b0 = models.efficientnet_b0() ``` - torchvision 0.11.0 requires version above 1.9.0. - However, problem arose in inference.py when using torch version above 1.8.0 https://github.com/snoop2head/ELimNet/blob/590152df6fd0cfa63718af581ca7c2d1956acb50/inference.py#L22-L25 - Currently using torch==1.7.0 and torchvision==0.10.0 as a workaround
https://github.com/snoop2head/ELimNet/blob/590152df6fd0cfa63718af581ca7c2d1956acb50/inference.py#L22-L25