# Summary of commands - uncomment one and run this script #### For the datasets in sections marked as "Automatic Download", dataset will be downloaded automatically downloaded before training begins. For "Manual Download", it is expected that it is manually downloaded and kept in the folder specified agaianst the --data_path option. ## ===================================================================================== ## Training ## ===================================================================================== ## Cifar100 Classification (Automatic Download) #### Training with MobileNetV2 #python ./scripts/train_classification_main.py --dataset_name cifar100_classification --model_name mobilenetv2_tv_x1 --data_path ./data/datasets/cifar100_classification --img_resize 32 --img_crop 32 --rand_scale 0.5 1.0 --strides 1 1 1 2 2 ## Cifar10 Classification (Automatic Download) #### Training with MobileNetV2 #python ./scripts/train_classification_main.py --dataset_name cifar10_classification --model_name mobilenetv2_tv_x1 --data_path ./data/datasets/cifar10_classification --img_resize 32 --img_crop 32 --rand_scale 0.5 1.0 --strides 1 1 1 2 2 ## ImageNet Classification (Automatic Download) #### Training with MobileNetV2 #python ./scripts/train_classification_main.py --dataset_name imagenet_classification --model_name mobilenetv2_tv_x1 --data_path ./data/datasets/imagenet_classification ## ImageNet Classification (Manual Download) #### Training with MobileNetV2 #python ./scripts/train_classification_main.py --dataset_name image_folder_classification --model_name mobilenetv2_tv_x1 --data_path ./data/datasets/image_folder_classification #### Training with MobileNetV2 - Small Resolution #python ./scripts/train_classification_main.py --dataset_name image_folder_classification --model_name mobilenetv2_tv_x1 --data_path ./data/datasets/image_folder_classification --img_resize 146 --img_crop 128 --batch_size 1024 --lr 0.2 --workers 16 #### Training with MobileNetV2 with 2x channels and expansion factor of 2 #python ./scripts/train_classification_main.py --dataset_name image_folder_classification --model_name mobilenetv2_tv_x2_t2 --data_path ./data/datasets/image_folder_classification --batch_size 256 ### Training with ResNet50 #python ./scripts/train_classification_main.py --dataset_name image_folder_classification --model_name resnet50_x1 --data_path ./data/datasets/image_folder_classification ### Training with ResNet50 - with half the number of channels - so roughly 1/4 the complexity #python ./scripts/train_classification_main.py --dataset_name image_folder_classification --model_name resnet50_xp5 --data_path ./data/datasets/image_folder_classification ## ===================================================================================== ## Validation ## ===================================================================================== #### cifar100 Validation - populate the pretrained model path below in ?? #python ./scripts/train_classification_main.py --phase validation --dataset_name cifar100_classification --model_name mobilenetv2_tv_x1 --data_path ./data/datasets/cifar100_classification --img_resize 32 --img_crop 32 \ #--pretrained=??? #### cifar10 Validation - populate the pretrained model path below in ?? #python ./scripts/train_classification_main.py --phase validation --dataset_name cifar10_classification --model_name mobilenetv2_tv_x1 --data_path ./data/datasets/cifar10_classification --img_resize 32 --img_crop 32 \ #--pretrained=??? #### Validation - populate the pretrained model path below in ?? or use https://download.pytorch.org/models/resnet50-19c8e357.pth for resnet50_x1 #python ./scripts/train_classification_main.py --phase validation --dataset_name image_folder_classification --model_name resnet50_x1 --data_path ./data/datasets/image_folder_classification \ #--pretrained https://download.pytorch.org/models/resnet50-19c8e357.pth #### Validation - populate the pretrained model path below in ?? or use https://download.pytorch.org/models/mobilenet_v2-b0353104.pth for mobilenetv2_tv_x1 #python ./scripts/train_classification_main.py --phase validation --dataset_name image_folder_classification --model_name mobilenetv2_tv_x1 --data_path ./data/datasets/image_folder_classification \ #--pretrained https://download.pytorch.org/models/mobilenet_v2-b0353104.pth #### Validation - populate the pretrained model path below in ?? for resnet50_xp5 #python ./scripts/train_classification_main.py --phase validation --dataset_name image_folder_classification --model_name resnet50_xp5 --data_path ./data/datasets/image_folder_classification \ #--pretrained ./data/modelzoo/pytorch/image_classification/imagenet1k/jacinto_ai/resnet50-0.5_2018-07-23_12-10-23.pth