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