Error importing from torchgeo.trainer import SegmentationTask

[lcoandrade]

I was about to run my Kaggle notebook again now and when I do from torchgeo.trainers import SemanticSegmentationTask I get:

---------------------------------------------------------------------------
ImportError                               Traceback (most recent call last)
Cell In[2], line 31
     29 from torchgeo.datamodules import GeoDataModule
     30 from torchgeo.datasets.splits import random_bbox_assignment
---> 31 from torchgeo.trainers import SemanticSegmentationTask
     33 from rasterio.plot import show
     34 from rasterio.merge import merge

File /opt/conda/lib/python3.10/site-packages/torchgeo/trainers/__init__.py:10
      8 from .classification import ClassificationTask, MultiLabelClassificationTask
      9 from .detection import ObjectDetectionTask
---> 10 from .moco import MoCoTask
     11 from .regression import PixelwiseRegressionTask, RegressionTask
     12 from .segmentation import SemanticSegmentationTask

File /opt/conda/lib/python3.10/site-packages/torchgeo/trainers/moco.py:17
     15 import torch.nn as nn
     16 import torch.nn.functional as F
---> 17 from lightly.loss import NTXentLoss
     18 from lightly.models.modules import MoCoProjectionHead
     19 from lightly.models.utils import deactivate_requires_grad, update_momentum

File /opt/conda/lib/python3.10/site-packages/lightly/loss/__init__.py:12
     10 from lightly.loss.msn_loss import MSNLoss
     11 from lightly.loss.negative_cosine_similarity import NegativeCosineSimilarity
---> 12 from lightly.loss.ntx_ent_loss import NTXentLoss
     13 from lightly.loss.pmsn_loss import PMSNCustomLoss, PMSNLoss
     14 from lightly.loss.swav_loss import SwaVLoss

File /opt/conda/lib/python3.10/site-packages/lightly/loss/ntx_ent_loss.py:12
      9 from torch import distributed as torch_dist
     10 from torch import nn
---> 12 from lightly.models.modules.memory_bank import MemoryBankModule
     13 from lightly.utils import dist
     16 class NTXentLoss(MemoryBankModule):

File /opt/conda/lib/python3.10/site-packages/lightly/models/__init__.py:22
     18 # Copyright (c) 2020. Lightly AG and its affiliates.
     19 # All Rights Reserved
     21 from lightly.models import utils
---> 22 from lightly.models.barlowtwins import BarlowTwins
     23 from lightly.models.byol import BYOL
     24 from lightly.models.moco import MoCo

File /opt/conda/lib/python3.10/site-packages/lightly/models/barlowtwins.py:13
     10 import torch
     11 import torch.nn as nn
---> 13 from lightly.models.modules import BarlowTwinsProjectionHead
     16 class BarlowTwins(nn.Module):
     17     """Implementation of BarlowTwins[0] network.
     18 
     19     Recommended loss: :py:class:`lightly.loss.barlow_twins_loss.BarlowTwinsLoss`
   (...)
     35 
     36     """

File /opt/conda/lib/python3.10/site-packages/lightly/models/modules/__init__.py:41
     38 if _timm_available:
     39     # Requires timm.
     40     from lightly.models.modules.heads_timm import AIMPredictionHead
---> 41     from lightly.models.modules.masked_causal_vision_transformer import (
     42         MaskedCausalVisionTransformer,
     43     )

File /opt/conda/lib/python3.10/site-packages/lightly/models/modules/masked_causal_vision_transformer.py:5
      3 import torch
      4 import torch.nn.functional as F
----> 5 from timm.layers import LayerType, Mlp, PatchEmbed
      6 from timm.models import _manipulate
      7 from timm.models.vision_transformer import Attention, Block, VisionTransformer

ImportError: cannot import name 'LayerType' from 'timm.layers' (/opt/conda/lib/python3.10/site-packages/timm/layers/__init__.py)

Yesterday was working fine. What could be the reason?

[adamjstewart]

Duplicate of #1824. Current workaround is to use:

pip install torchgeo lightly!=1.4.26

to install a version of lightly that's compatible with smp.