Fine Tuning Option

[ndisci]

Hi, Is there any option to fine-tune the pre-trained models ? Thanks for your time.

[Rikorose]

Hi,

This is not implemented by the framework. The usual process is straight forward: Load the pretrained weights for all but the last DNN layers. Then, do your normal training on your fine-tune data.

[github-actions[bot]]

This issue is stale because it has been open 90 days with no activity. Remove stale label or comment or this will be closed in 7 days.

[youssefabdelm]

Hi,

This is not implemented by the framework. The usual process is straight forward: Load the pretrained weights for all but the last DNN layers. Then, do your normal training on your fine-tune data.

Hey @Rikorose I'd love to finetune on the existing pretrained weights as well. Could you provide a more detailed description or examples on how to go about doing the "load pretrained weights all but for last DNN layers" step?

The original model is excellent so would love to be able to use that, I just want to experiment with my own noise types to see if it can recover some lost aspects.

No worries if too lengthy/much of a hassle though.