cleanedup STE for QAT. Added RegNetX models
authorManu Mathew <a0393608@ti.com>
Thu, 6 Aug 2020 19:03:01 +0000 (00:33 +0530)
committerManu Mathew <a0393608@ti.com>
Thu, 6 Aug 2020 19:04:18 +0000 (00:34 +0530)
commita3a9f4a2a56d6a623bfe052e5d052cdea7b19497
treed0436e781306426c5d2d0e8f9e759e830b75b034
parent3995b7ee0d97148874e9c9f499b4c92eb5432c62
cleanedup STE for QAT. Added RegNetX models
41 files changed:
docs/Image_Classification.md
docs/Semantic_Segmentation.md
modules/pytorch_jacinto_ai/engine/train_classification.py
modules/pytorch_jacinto_ai/engine/train_pixel2pixel.py
modules/pytorch_jacinto_ai/xnn/layers/__init__.py
modules/pytorch_jacinto_ai/xnn/layers/activation.py
modules/pytorch_jacinto_ai/xnn/layers/function.py
modules/pytorch_jacinto_ai/xnn/layers/functional.py
modules/pytorch_jacinto_ai/xnn/layers/quant_ste.py [new file with mode: 0644]
modules/pytorch_jacinto_ai/xnn/layers/resize_blocks.py
modules/pytorch_jacinto_ai/xnn/layers/rf_blocks.py
modules/pytorch_jacinto_ai/xnn/quantize/quant_base_module.py
modules/pytorch_jacinto_ai/xnn/quantize/quant_calib_module.py
modules/pytorch_jacinto_ai/xnn/quantize/quant_graph_module.py
modules/pytorch_jacinto_ai/xnn/quantize/quant_test_module.py
modules/pytorch_jacinto_ai/xnn/quantize/quant_train_module.py
modules/pytorch_jacinto_ai/xnn/utils/__init__.py
modules/pytorch_jacinto_ai/xnn/utils/count_flops.py
modules/pytorch_jacinto_ai/xnn/utils/data_utils.py [new file with mode: 0644]
modules/pytorch_jacinto_ai/xnn/utils/depth_utils.py [new file with mode: 0644]
modules/pytorch_jacinto_ai/xnn/utils/function_utils.py [new file with mode: 0644]
modules/pytorch_jacinto_ai/xnn/utils/hist_utils.py [new file with mode: 0644]
modules/pytorch_jacinto_ai/xnn/utils/load_weights.py
modules/pytorch_jacinto_ai/xnn/utils/range_utils.py [new file with mode: 0644]
modules/pytorch_jacinto_ai/xnn/utils/tensor_utils.py
modules/pytorch_jacinto_ai/xvision/models/classification/__init__.py
modules/pytorch_jacinto_ai/xvision/models/multi_input_net.py
modules/pytorch_jacinto_ai/xvision/models/pixel2pixel/deeplabv3lite.py
modules/pytorch_jacinto_ai/xvision/models/pixel2pixel/fpnlite_pixel2pixel.py
modules/pytorch_jacinto_ai/xvision/models/pixel2pixel/pixel2pixelnet_utils.py
modules/pytorch_jacinto_ai/xvision/models/pixel2pixel/unetlite_pixel2pixel.py
modules/pytorch_jacinto_ai/xvision/models/regnet.py [new file with mode: 0644]
modules/pytorch_jacinto_ai/xvision/transforms/image_transform_utils.py
modules/pytorch_jacinto_ai/xvision/transforms/image_transforms.py
modules/pytorch_jacinto_ai/xvision/transforms/transforms.py
requirements.txt
run_classification.sh
run_quantization.sh
run_segmentation.sh
scripts/train_classification_main.py
scripts/train_segmentation_main.py