File size: 635 Bytes
c587e44 |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
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 |