Skip to content

Script for training BERT model doesn't save the model... #1

@illided

Description

@illided

Script run_finetune_kaznerd.py doesn't actually save the model, which makes it impossible to use for inference.
Can be fixed with simple trainer.save(model_path) right after the trainer.train()

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions