D-Psyntom / README.md
eperezs's picture
Update README.md
4c649b3 verified
|
raw
history blame
509 Bytes
metadata
library_name: transformers
tags: []

Model Card for Model ID

image/jpeg

---------

checkpoint = "mental/mental-bert-base-uncased"

learning_rate = 2e-5

batch_size = 16

num_train_epochs = 5

weight_decay = 0.01

metric_name = "accuracy"

optimizer = AdamW

test_size = 0.2

logging_steps = 25

column_item = "labels_array"