Skip to content

Commit

Permalink
Fix typo bug
Browse files Browse the repository at this point in the history
  • Loading branch information
BloodAxe committed Mar 10, 2024
1 parent 1634a3b commit 594b299
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion pytorch_toolbelt/modules/encoders/timm/hrnet.py
Original file line number Diff line number Diff line change
Expand Up @@ -73,7 +73,7 @@ def __init__(

class TimmHRNetW18SmallV2Encoder(HRNetTimmEncoder):
def __init__(
self, elf, pretrained=True, use_incre_features: bool = True, layers=None, first_conv_stride_one: bool = False
self, pretrained=True, use_incre_features: bool = True, layers=None, first_conv_stride_one: bool = False
):
from timm.models import hrnet

Expand Down

0 comments on commit 594b299

Please sign in to comment.