File size: 492 Bytes
2e8434f |
1 2 3 4 5 6 7 8 9 10 |
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=24, shuffle=True) for epoch in range(3): Epoch 0: 100%|██████████| 83/83 [01:14<00:00, 1.12it/s, loss=2.14] Epoch 1: 100%|██████████| 83/83 [01:14<00:00, 1.12it/s, loss=1.04] Epoch 2: 100%|██████████| 83/83 [01:14<00:00, 1.12it/s, loss=0.68] |