This app has been developed towards continous asessment of CS7GV1-A-Y-201718 (COMPUTER VISION) module.
The dataset used here is Tiny ImageNet.
python Main.py
| Command Line Args | Values |
|---|---|
| --cuda | True / False |
| --datapath | Source of data |
| --aug | Augmentation strategy to be used (H_FLIP, RANDOM_CROP_FLIP) |
| --model | Model to be used for training |
| --pretrained | True / False |
| --batch_size | Batch Size |
| --epochs | No of epochs |
| --tag | Save this model with tag name |