pytorch-ner
pytorch-ner copied to clipboard
Pipeline for training NER models using PyTorch.
Add model checkpoint This commit adds checkpoint.py file which allows: 1. Saving weights based on lowest val_loss from flag save_best_weights in config.yml 2. Saving weights frequently with specified frequency int...
Adding Early Stopping Feature This commits adds early stopping feature, early stopping class is created by Bjarten/early-stopping-pytorch [link below] Added several variables in config.yml: 1. early_stopping_flag: wether to enable early_stopping...
add config parametrization for: - full path to modules - neural_network architecture (BiLSTM, BiLSTMAttn, ...) - embedding (pre-trained or not) - embedding dropout (SpatialDropout, WordEmbeddingsDropout) - attention - ...
add config parametrization for: - full path to modules - neural_network architecture (BiLSTM, BiLSTMAttn, ...) - embedding (pre-trained or not) - embedding dropout (SpatialDropout, WordEmbeddingsDropout) - attention - ...