add vit_base
Showing
- research/cv/vit_base/README_CN.md 313 additions, 0 deletionsresearch/cv/vit_base/README_CN.md
- research/cv/vit_base/ascend310_infer/CMakeLists.txt 15 additions, 0 deletionsresearch/cv/vit_base/ascend310_infer/CMakeLists.txt
- research/cv/vit_base/ascend310_infer/build.sh 23 additions, 0 deletionsresearch/cv/vit_base/ascend310_infer/build.sh
- research/cv/vit_base/ascend310_infer/inc/utils.h 32 additions, 0 deletionsresearch/cv/vit_base/ascend310_infer/inc/utils.h
- research/cv/vit_base/ascend310_infer/src/main.cc 162 additions, 0 deletionsresearch/cv/vit_base/ascend310_infer/src/main.cc
- research/cv/vit_base/ascend310_infer/src/utils.cc 159 additions, 0 deletionsresearch/cv/vit_base/ascend310_infer/src/utils.cc
- research/cv/vit_base/eval.py 89 additions, 0 deletionsresearch/cv/vit_base/eval.py
- research/cv/vit_base/export.py 62 additions, 0 deletionsresearch/cv/vit_base/export.py
- research/cv/vit_base/postprocess.py 45 additions, 0 deletionsresearch/cv/vit_base/postprocess.py
- research/cv/vit_base/preprocess.py 38 additions, 0 deletionsresearch/cv/vit_base/preprocess.py
- research/cv/vit_base/requirements.txt 1 addition, 0 deletionsresearch/cv/vit_base/requirements.txt
- research/cv/vit_base/scripts/run_distribution_train_ascend.sh 43 additions, 0 deletions...arch/cv/vit_base/scripts/run_distribution_train_ascend.sh
- research/cv/vit_base/scripts/run_infer_310.sh 121 additions, 0 deletionsresearch/cv/vit_base/scripts/run_infer_310.sh
- research/cv/vit_base/scripts/run_standalone_eval_ascend.sh 21 additions, 0 deletionsresearch/cv/vit_base/scripts/run_standalone_eval_ascend.sh
- research/cv/vit_base/scripts/run_standalone_train_ascend.sh 21 additions, 0 deletionsresearch/cv/vit_base/scripts/run_standalone_train_ascend.sh
- research/cv/vit_base/src/config.py 53 additions, 0 deletionsresearch/cv/vit_base/src/config.py
- research/cv/vit_base/src/dataset.py 80 additions, 0 deletionsresearch/cv/vit_base/src/dataset.py
- research/cv/vit_base/src/modeling_ms.py 221 additions, 0 deletionsresearch/cv/vit_base/src/modeling_ms.py
- research/cv/vit_base/src/net_config.py 117 additions, 0 deletionsresearch/cv/vit_base/src/net_config.py
- research/cv/vit_base/train.py 211 additions, 0 deletionsresearch/cv/vit_base/train.py
Please register or sign in to comment