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