search

voldemortX / pytorch-auto-drive

PytorchAutoDrive: Segmentation models (ERFNet, ENet, DeepLab, FCN...) and Lane detection models (SCNN, RESA, LSTR, LaneATT, BézierLaneNet...) based on PyTorch with fast training, visualization, benchmarking & deployment help
BSD 3-Clause "New" or "Revised" License
854 stars 138 forks source link

Support PyTorch 1.7+ #141

Closed voldemortX closed 1 year ago

voldemortX commented 1 year ago

After the SCNN & RESA support of PyTorch >= 1.8 in #139.

This PR bump timm version from 0.3.2 to 0.4.5 to fully support PyTorch 1.7+. Previously, there is an error like this:

cannot import name 'container_abcs' from 'torch._six'

reference: https://github.com/huggingface/pytorch-image-models/pull/421