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