mnist-student / config.json
lbernick's picture
Upload trained MNIST student model
eda3930 verified
{
"architecture": "ConvNet",
"task": "MNIST classification",
"batch_size": 128,
"epochs": 10,
"learning_rate": 0.001,
"weight_decay": 0.0,
"trainset_size": 50000,
"seed": 42,
"device": "cuda",
"test_accuracy": 0.9733,
"test_loss": 0.09284641495328397
}