File size: 492 Bytes
a25f33b
 
 
 
 
 
 
 
 
 
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]