train_model(train_batches, 20, model, opt, X_val_aug, y_val_one_hot, calc_device)
Epoch: 1
Training Loss: 1.18 Training Acc: 0.38 Validation Loss: 1.13 Validation Acc: 0.51
Epoch: 2
Training Loss: 1.07 Training Acc: 0.43 Validation Loss: 1.04 Validation Acc: 0.42
Epoch: 3
Training Loss: 1.03 Training Acc: 0.47 Validation Loss: 0.98 Validation Acc: 0.48
Epoch: 4
Training Loss: 1.02 Training Acc: 0.45 Validation Loss: 0.96 Validation Acc: 0.5
Epoch: 5
Training Loss: 1.01 Training Acc: 0.47 Validation Loss: 0.97 Validation Acc: 0.48
Epoch: 6
Training Loss: 0.99 Training Acc: 0.51 Validation Loss: 0.96 Validation Acc: 0.49
Epoch: 7
Training Loss: 0.99 Training Acc: 0.49 Validation Loss: 0.94 Validation Acc: 0.56
Epoch: 8
Training Loss: 0.98 Training Acc: 0.52 Validation Loss: 0.95 Validation Acc: 0.49
Epoch: 9
Training Loss: 0.97 Training Acc: 0.51 Validation Loss: 0.93 Validation Acc: 0.52
Epoch: 10
Training Loss: 0.96 Training Acc: 0.53 Validation Loss: 0.93 Validation Acc: 0.53
Epoch: 11
Training Loss: 0.94 Training Acc: 0.55 Validation Loss: 0.92 Validation Acc: 0.52
Epoch: 12
Training Loss: 0.93 Training Acc: 0.57 Validation Loss: 0.91 Validation Acc: 0.54
Epoch: 13
Training Loss: 0.91 Training Acc: 0.57 Validation Loss: 0.93 Validation Acc: 0.53
Epoch: 14
Training Loss: 0.9 Training Acc: 0.58 Validation Loss: 0.92 Validation Acc: 0.53
Epoch: 15
Training Loss: 0.89 Training Acc: 0.58 Validation Loss: 0.86 Validation Acc: 0.56
Epoch: 16
Training Loss: 0.86 Training Acc: 0.61 Validation Loss: 0.83 Validation Acc: 0.62
Epoch: 17
Training Loss: 0.82 Training Acc: 0.66 Validation Loss: 0.81 Validation Acc: 0.65
Epoch: 18
Training Loss: 0.8 Training Acc: 0.66 Validation Loss: 0.69 Validation Acc: 0.74
Epoch: 19
Training Loss: 0.78 Training Acc: 0.69 Validation Loss: 0.64 Validation Acc: 0.82
Epoch: 20
Training Loss: 0.71 Training Acc: 0.72 Validation Loss: 0.6 Validation Acc: 0.74