Skip to content

Commit e1678c6

Browse files
committed
fix head bug
1 parent ca03c21 commit e1678c6

File tree

1 file changed

+1
-1
lines changed
  • change_detection_pytorch/upernet

1 file changed

+1
-1
lines changed

change_detection_pytorch/upernet/model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
from typing import Optional, Union
22

3-
from ..base import ClassificationHead, SegmentationHead, SegmentationModel, SegmentationOCRHead
3+
from ..base import ClassificationHead, SegmentationHead, SegmentationModel
44
from ..encoders import get_encoder
55
from .decoder import UPerNetDecoder
66

0 commit comments

Comments
 (0)