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