quantization_example - RandomSampler is used when epoch_size!=0. epoch_size=0.1 means...
[jacinto-ai/pytorch-jacinto-ai-devkit.git] / run_depth.sh
1 # Summary of commands - uncomment one and run this script
2 #### 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.
4 ## =====================================================================================
5 ## Training
6 ## =====================================================================================
7 #### KITTI Depth (Manual Download) - Training with MobileNetV2+DeeplabV3Lite
8 #python ./scripts/train_depth_main.py --dataset_name kitti_depth --model_name deeplabv3lite_mobilenetv2_tv --data_path ./data/datasets/kitti/kitti_depth/data --img_resize 384 768 --output_size 374 1242 \
9 #--pretrained https://download.pytorch.org/models/mobilenet_v2-b0353104.pth
11 #### KITTI Depth (Manual Download) - Training with ResNet50+FPN
12 #python ./scripts/train_depth_main.py --dataset_name kitti_depth --model_name fpnlite_pixel2pixel_aspp_resnet50 --data_path ./data/datasets/kitti/kitti_depth/data --img_resize 384 768 --output_size 374 1242 \
13 #--pretrained https://download.pytorch.org/models/resnet50-19c8e357.pth
15 ## =====================================================================================
16 ## Validation
17 ## =====================================================================================
18 #### KITTI Depth (Manual Download) - Validation - populate the pretrained path in ??
19 #python ./scripts/train_depth_main.py --phase validation --dataset_name kitti_depth --model_name deeplabv3lite_mobilenetv2_tv --data_path ./data/datasets/kitti/kitti_depth/data --img_resize 384 768 --output_size 374 1242 \
20 #--pretrained=???
22 #### KITTI Depth (Manual Download) - Inference - populate the pretrained filename in ??
23 #python ./scripts/infer_depth_main.py --dataset_name kitti_depth --model_name deeplabv3lite_mobilenetv2_tv --data_path ./data/datasets/kitti/kitti_depth/data --img_resize 384 768 --output_size 374 1242 \
24 #--pretrained ???