Section4

hf-notebookseschapter3course

Entrenamiento completo

Install the Transformers, Datasets, and Evaluate libraries to run this notebook.

[ ]
[ ]
[ ]
[ ]
[ ]
[ ]
{'attention_mask': torch.Size([8, 65]),
, 'input_ids': torch.Size([8, 65]),
, 'labels': torch.Size([8]),
, 'token_type_ids': torch.Size([8, 65])}
[ ]
[ ]
tensor(0.5441, grad_fn=<NllLossBackward>) torch.Size([8, 2])
[ ]
[ ]
1377
[ ]
device(type='cuda')
[ ]
[ ]
{'accuracy': 0.8431372549019608, 'f1': 0.8907849829351535}
[ ]
[ ]
[ ]