This classifier was trained using the
auto-cats-and-dogs dataset. It was trained over 5 epochs using a pretrained resent18.
config = {
"batch_size": 64,
"num_epochs": 5,
"lr": 0.005,
"betas": (0.9, 0.999),
"eps": 1e-6,
"lr": 8e-3,
"do_eval": True
}
We can see in the figures below the training plots for accuracy and the loss in both, training and validation sets.
A batch with some missclassifications can be seen in the picture below.
Results
