Skip to content

Commit

Permalink
Fix import statement for MultiLevelNeck in DOFASeg model
Browse files Browse the repository at this point in the history
  • Loading branch information
valhassan committed Nov 25, 2024
1 parent 814cd10 commit 4f842ba
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion geo_deep_learning/models/dofa/dofa_seg.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@
import torch.nn as nn
import torch.nn.functional as F
import torch.nn.init as init
from neckhead import MultiLevelNeck
from .neckhead import MultiLevelNeck
from pathlib import Path
from timm.models.vision_transformer import Block
from torch import Tensor
Expand Down

0 comments on commit 4f842ba

Please sign in to comment.