Skip to content

Transfer learning #2

@mrocamora

Description

@mrocamora

Another feature we would need is transfer learning, for adapting a model to our particular purposes.

In particular, we would like to train a model, for instance, SB_CNN with URBAN_SED. Then, we would like to remove the last fully connected layers and replace them with new fully connected layers. Finally, we would like to train these layers with a certain dataset (e.g. recordings of the sound classes we want to detect), while keeping the weights fixed for the rest of the model.

I am willing to discuss the different possible ways of implementing this, if you agree.

Thanks!

Metadata

Metadata

Assignees

No one assigned

    Labels

    enhancementNew feature or request

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions