!330 [社区模型征集] Add GPU support ntsnet
Merge pull request !330 from ZJUTER0126/ntsnet
No related branches found
No related tags found
Showing
- research/cv/ntsnet/README.md 52 additions, 29 deletionsresearch/cv/ntsnet/README.md
- research/cv/ntsnet/data_prepare.py 69 additions, 0 deletionsresearch/cv/ntsnet/data_prepare.py
- research/cv/ntsnet/eval.py 21 additions, 4 deletionsresearch/cv/ntsnet/eval.py
- research/cv/ntsnet/export.py 5 additions, 4 deletionsresearch/cv/ntsnet/export.py
- research/cv/ntsnet/scripts/run_distribute_train_ascend.sh 2 additions, 2 deletionsresearch/cv/ntsnet/scripts/run_distribute_train_ascend.sh
- research/cv/ntsnet/scripts/run_distribute_train_gpu.sh 53 additions, 0 deletionsresearch/cv/ntsnet/scripts/run_distribute_train_gpu.sh
- research/cv/ntsnet/scripts/run_eval_ascend.sh 72 additions, 0 deletionsresearch/cv/ntsnet/scripts/run_eval_ascend.sh
- research/cv/ntsnet/scripts/run_eval_gpu.sh 11 additions, 11 deletionsresearch/cv/ntsnet/scripts/run_eval_gpu.sh
- research/cv/ntsnet/scripts/run_standalone_train_ascend.sh 8 additions, 3 deletionsresearch/cv/ntsnet/scripts/run_standalone_train_ascend.sh
- research/cv/ntsnet/scripts/run_standalone_train_gpu.sh 65 additions, 0 deletionsresearch/cv/ntsnet/scripts/run_standalone_train_gpu.sh
- research/cv/ntsnet/src/callback.py 58 additions, 0 deletionsresearch/cv/ntsnet/src/callback.py
- research/cv/ntsnet/src/config.py 32 additions, 5 deletionsresearch/cv/ntsnet/src/config.py
- research/cv/ntsnet/src/dataset.py 43 additions, 15 deletionsresearch/cv/ntsnet/src/dataset.py
- research/cv/ntsnet/src/lr_generator.py 26 additions, 3 deletionsresearch/cv/ntsnet/src/lr_generator.py
- research/cv/ntsnet/src/network.py 8 additions, 8 deletionsresearch/cv/ntsnet/src/network.py
- research/cv/ntsnet/train.py 107 additions, 38 deletionsresearch/cv/ntsnet/train.py
Please register or sign in to comment