israfelsr's picture
Create README.md
07f5bd8
|
raw
history blame
445 Bytes

UpsideDownClassifier

This classifier was trained using the auto-cats-and-dogs dataset. It was trained over 5 epochs using a pretrained resent18.

The configuration for the model was

config = {
    "batch_size": 64,
    "num_epochs": 5,
    "lr": 0.005,
    "betas": (0.9, 0.999),
    "eps": 1e-6,
    "lr": 8e-3,
    "do_eval": True
}

Traning Plots

Some Results