{ | |
"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." | |
} | |