Skip to content

error raised when loading model_weights "include_top=True" #11

@TwistFateBOY

Description

@TwistFateBOY

here is the code
model = VGGish(include_top=True)
model.load_weights(WEIGHTS_PATH_TOP)

here is the bug
ValueError: Dimension 0 in both shapes must be equal, but are 63488 and 12288. Shapes are [63488,4096] and [12288,4096]. for 'Assign_12' (op: 'Assign') with input shapes: [63488,4096], [12288,4096].

what's wrong?

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