File size: 595 Bytes
9a437fa |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
{
"model_type": "TinyVGG",
"num_epochs": 10,
"batch_size": 32,
"learning_rate": 0.001,
"input_shape": 3,
"hidden_units": 64,
"output_shape": 10,
"kernel_size": 3,
"stride": 1,
"padding": 0,
"pooling_kernel_size": 2,
"pooling_stride": 2,
"framework": "PyTorch",
"image_size": 64,
"author": "Ajit",
"description": "TinyVGG model for image classification with 64x64 input images."
}
|