Skip to content

Use Callback for 2 stage training in DeeBert #34

Description

@JulesBelveze

DeeBert models need to be fine-tuned in a two step fashion: first the final layer and then the ramps.
The current implementation requires the user to do two different training. However, this can be achieved in one-shot using a pl.Callback, as done for TheseusBert.

Metadata

Metadata

Assignees

Labels

enhancementNew feature or request

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions