Skip to content

Custom # of channels from new PR not working #93

@s-rog

Description

@s-rog
EfficientNet.from_pretrained("efficientnet-b{}".format(bN),
                                               num_classes=6,
                                               in_channels=channels)
TypeError: from_pretrained() got an unexpected keyword argument 'in_channels'

might have to do with the fact that there are 2 from_pretrained() methods? not exactly sure

Edit: I made some changes in my fork, will post a PR once I have time to test it

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions