|
optim = AdamW(model.parameters(), lr=5e-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(7): |
|
Epoch 0: 100%|鈻堚枅鈻堚枅鈻堚枅鈻堚枅鈻堚枅| 94/94 [01:18<00:00, 1.19it/s, loss=2.38] |
|
Epoch 1: 100%|鈻堚枅鈻堚枅鈻堚枅鈻堚枅鈻堚枅| 94/94 [01:18<00:00, 1.19it/s, loss=1.9] |
|
Epoch 2: 100%|鈻堚枅鈻堚枅鈻堚枅鈻堚枅鈻堚枅| 94/94 [01:18<00:00, 1.19it/s, loss=1.56] |
|
Epoch 3: 100%|鈻堚枅鈻堚枅鈻堚枅鈻堚枅鈻堚枅| 94/94 [01:18<00:00, 1.19it/s, loss=1.44] |
|
Epoch 4: 100%|鈻堚枅鈻堚枅鈻堚枅鈻堚枅鈻堚枅| 94/94 [01:18<00:00, 1.19it/s, loss=1.52] |
|
Epoch 5: 100%|鈻堚枅鈻堚枅鈻堚枅鈻堚枅鈻堚枅| 94/94 [01:18<00:00, 1.19it/s, loss=0.492] |
|
Epoch 6: 100%|鈻堚枅鈻堚枅鈻堚枅鈻堚枅鈻堚枅| 94/94 [01:18<00:00, 1.19it/s, loss=0.195] |
|
|
|
Precisi贸n del modelo ajustado: 0.7771084337349398 |
|
|