Skip to content

Transfer EfficientNet_v1 problem!  #1

@Greatpep

Description

@Greatpep

Firstly thanks for your brilliant model implement!
But when I try to transfer EfficientNet_v1 with efficientnet-b1_imagenet_1000.h5, there is a bug said "Layer #1 (named "conv2d_1") expects 2 weight(s), but the saved weights have 1 element(s)."
I load the checkpoint by model.load_weights(TRANSFER_MODEL_NAME, by_name = True)
The checkpoint was downloaded from https://github.com/qubvel/efficientnet/releases/download/v0.0.1/efficientnet-b1_imagenet_1000.h5
I know the which layer the bug come from but can not understand why this conv2D layer expect 2 weights and does the problem comes of the checkpoint or the code?

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