| 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(5): |
|
|
| Epoch 0: 100%|鈻堚枅鈻堚枅鈻堚枅鈻堚枅鈻堚枅| 16/16 [00:10<00:00, 1.59it/s, loss=2.55] |
| Epoch 1: 100%|鈻堚枅鈻堚枅鈻堚枅鈻堚枅鈻堚枅| 16/16 [00:10<00:00, 1.59it/s, loss=1.5] |
| Epoch 2: 100%|鈻堚枅鈻堚枅鈻堚枅鈻堚枅鈻堚枅| 16/16 [00:10<00:00, 1.59it/s, loss=0.783] |
| Epoch 3: 100%|鈻堚枅鈻堚枅鈻堚枅鈻堚枅鈻堚枅| 16/16 [00:10<00:00, 1.59it/s, loss=0.407] |
| Epoch 4: 100%|鈻堚枅鈻堚枅鈻堚枅鈻堚枅鈻堚枅| 16/16 [00:10<00:00, 1.59it/s, loss=0.126] |
|
|
| Precisi贸n del modelo ajustado: 0.9232283464566929 |
|
|
