!2817 Implemented training and evaluation of STGCN on a GPU
Merge pull request !2817 from Marina Molchanova/stgcn_gpu
No related branches found
No related tags found
Showing
- research/cv/stgcn/README.md 313 additions, 0 deletionsresearch/cv/stgcn/README.md
- research/cv/stgcn/README_CN.md 136 additions, 92 deletionsresearch/cv/stgcn/README_CN.md
- research/cv/stgcn/eval.py 145 additions, 0 deletionsresearch/cv/stgcn/eval.py
- research/cv/stgcn/export.py 14 additions, 29 deletionsresearch/cv/stgcn/export.py
- research/cv/stgcn/modelarts/start_train.py 2 additions, 2 deletionsresearch/cv/stgcn/modelarts/start_train.py
- research/cv/stgcn/postprocess.py 5 additions, 14 deletionsresearch/cv/stgcn/postprocess.py
- research/cv/stgcn/preprocess.py 6 additions, 17 deletionsresearch/cv/stgcn/preprocess.py
- research/cv/stgcn/requirements.txt 4 additions, 0 deletionsresearch/cv/stgcn/requirements.txt
- research/cv/stgcn/scripts/run_distribute_train.sh 8 additions, 6 deletionsresearch/cv/stgcn/scripts/run_distribute_train.sh
- research/cv/stgcn/scripts/run_eval_ascend.sh 13 additions, 14 deletionsresearch/cv/stgcn/scripts/run_eval_ascend.sh
- research/cv/stgcn/scripts/run_eval_gpu.sh 37 additions, 0 deletionsresearch/cv/stgcn/scripts/run_eval_gpu.sh
- research/cv/stgcn/scripts/run_single_train_gpu.sh 67 additions, 0 deletionsresearch/cv/stgcn/scripts/run_single_train_gpu.sh
- research/cv/stgcn/src/argparser.py 55 additions, 0 deletionsresearch/cv/stgcn/src/argparser.py
- research/cv/stgcn/src/config.py 6 additions, 0 deletionsresearch/cv/stgcn/src/config.py
- research/cv/stgcn/src/dataloader.py 7 additions, 9 deletionsresearch/cv/stgcn/src/dataloader.py
- research/cv/stgcn/src/model/layers.py 42 additions, 26 deletionsresearch/cv/stgcn/src/model/layers.py
- research/cv/stgcn/src/model/metric.py 1 addition, 0 deletionsresearch/cv/stgcn/src/model/metric.py
- research/cv/stgcn/src/model/models.py 6 additions, 5 deletionsresearch/cv/stgcn/src/model/models.py
- research/cv/stgcn/src/utility.py 7 additions, 7 deletionsresearch/cv/stgcn/src/utility.py
- research/cv/stgcn/test.py 0 additions, 190 deletionsresearch/cv/stgcn/test.py
Please register or sign in to comment