tinyvgg / config.json
ajitsi's picture
adding config files
9a437fa
raw
history blame contribute delete
595 Bytes
{
"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."
}