|
optim = AdamW(model.parameters(), lr=3e-5) #tasa de aprendizaje |
|
# Se inicializa el cargador de datos para los datos de entrenamiento |
|
train_loader = DataLoader(train_dataset, batch_size=16, shuffle=True) |
|
|
|
for epoch in range(4): |
|
|
|
Epoch 0: 100%|鈻堚枅鈻堚枅鈻堚枅鈻堚枅鈻堚枅| 94/94 [01:58<00:00, 1.26s/it, loss=1.08] |
|
Epoch 1: 100%|鈻堚枅鈻堚枅鈻堚枅鈻堚枅鈻堚枅| 94/94 [02:00<00:00, 1.28s/it, loss=2.86] |
|
Epoch 2: 100%|鈻堚枅鈻堚枅鈻堚枅鈻堚枅鈻堚枅| 94/94 [02:00<00:00, 1.28s/it, loss=0.612] |
|
Epoch 3: 100%|鈻堚枅鈻堚枅鈻堚枅鈻堚枅鈻堚枅| 94/94 [02:00<00:00, 1.28s/it, loss=0.0787] |
|
|
|
Precisi贸n del modelo ajustado: 0.8043478260869565 |
|
|