File size: 1,859 Bytes
f792007 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 | ================================================== Training config ================================================== Model name : castorini/afriberta_base Tokenizer name : castorini/afriberta_base Number of labels : 22 Batch size : 8 Learning rate : 5e-05 Number of epochs : 16 Random seed : 42 ================================================== Epoch 1 ================================================== Traning loss : 818.2853 Validation loss : 0.1867 Precision : 0.8856 Recall : 0.8025 F1-score : 0.8285 Checkpoint saved ================================================== Epoch 2 ================================================== Traning loss : 245.6919 Validation loss : 0.1653 Precision : 0.9115 Recall : 0.7915 F1-score : 0.8174 ================================================== Epoch 3 ================================================== Traning loss : 124.5959 Validation loss : 0.1279 Precision : 0.9503 Recall : 0.8683 F1-score : 0.8944 Checkpoint saved ================================================== Epoch 4 ================================================== Traning loss : 76.5310 Validation loss : 0.2006 Precision : 0.9543 Recall : 0.8714 F1-score : 0.8966 Checkpoint saved ================================================== Epoch 5 ================================================== Traning loss : 55.0676 Validation loss : 0.1981 Precision : 0.9374 Recall : 0.8800 F1-score : 0.8927 ================================================== Epoch 6 ================================================== Traning loss : 36.3393 Validation loss : 0.1937 Precision : 0.9037 Recall : 0.8220 F1-score : 0.8429 ================================================== Epoch 7 ================================================== Traning loss : 21.6568 Validation loss : 0.1819 Precision : 0.8993 Recall : 0.8509 F1-score : 0.8652 Early stopping triggered |