lbernick's picture
Upload trained MNIST distilled student model
8fe7bd6 verified
Raw
History Blame Contribute Delete
362 Bytes
{
"architecture": "StudentNet",
"task": "MNIST classification",
"training_method": "knowledge_distillation",
"batch_size": 128,
"epochs": 10,
"learning_rate": 0.001,
"weight_decay": 0.0,
"trainset_size": 50000,
"temperature": 3.0,
"alpha": 0.5,
"seed": 42,
"device": "cuda",
"test_accuracy": 0.9785,
"test_loss": 0.08084478095881059
}