CODE 1: APPLY TECHNIQUE BATCH NORMALIZATION, REGULARIZATION, and GLOBAL AVERAGE POOLING to a Basic Skeleton
- Use nn.Sequential
- Add BatchNorm
- Apply dropOut on each layer
- Use GAP in last layer
- Parameters: 5.1k
- Best Training Accuracy: 93.31
- Best Test Accuracy: 97.78
- Model is way to much lighter
- It seems model is under-fitting cause of very less number of parameters
- Parameters: 10 k
- Best training accuracy = 99.88
- Best Test accuracy = 99.26%
- Parameters: 10 k
- Best training accuracy = 99.04
- Best Test accuracy = 99.13%
- Parameters: 5.1 k
- Best training accuracy = 93.91
- Best Test accuracy = 97.78%
- Increase model capacity at the end (add layer after GAP)
- Perform MaxPooling at RF=5 and using only one maxpooling layer
- Parameters: 7.9k
- Best Training Accuracy: 99.20%
- Best Test Accuracy: 99.39%
- Model is very good.
- No overfitting
- Still model is not able to get 99.4%
- Add rotation, of (-7 to 7) degrees.
- Add StepLR scheduler
- Parameters: 7.9k
- Best training accuracy = 99.30
- Best Test accuracy = 99.47%
- Model is awesome!!!
- No overfittng
- Target achieved