Views
No views yet
tokenizer = AutoTokenizer.from_pretrained("ltgoslo/norbert2")1training_args = TrainingArguments(
2 output_dir='./results', # output directory
3 num_train_epochs=10, # total number of training epochs
4 per_device_train_batch_size=16, # batch size per device during training
5 per_device_eval_batch_size=64, # batch size for evaluation
6 warmup_steps=500, # number of warmup steps for learning rate scheduler
7 weight_decay=0.01, # strength of weight decay
8 logging_dir='./logs', # directory for storing logs
9 logging_steps=10,
10)1{
2'accuracy': 0.8357214261912695,
3 'recall': 0.886873508353222,
4 'precision': 0.8789025543992431,
5 'f1': 0.8828700403896412,
6 'total_time_in_seconds': 94.33071640000003,
7 'samples_per_second': 31.81360340013276,
8 'latency_in_seconds': 0.03143309443518828
9 }