add gpu support on dcgan
Showing
- research/cv/dcgan/README.md 62 additions, 36 deletionsresearch/cv/dcgan/README.md
- research/cv/dcgan/eval.py 4 additions, 3 deletionsresearch/cv/dcgan/eval.py
- research/cv/dcgan/export.py 15 additions, 13 deletionsresearch/cv/dcgan/export.py
- research/cv/dcgan/gpu_infer/CMakeLists.txt 16 additions, 0 deletionsresearch/cv/dcgan/gpu_infer/CMakeLists.txt
- research/cv/dcgan/gpu_infer/build.sh 28 additions, 0 deletionsresearch/cv/dcgan/gpu_infer/build.sh
- research/cv/dcgan/gpu_infer/inc/utils.h 36 additions, 0 deletionsresearch/cv/dcgan/gpu_infer/inc/utils.h
- research/cv/dcgan/gpu_infer/src/main.cc 116 additions, 0 deletionsresearch/cv/dcgan/gpu_infer/src/main.cc
- research/cv/dcgan/gpu_infer/src/utils.cc 195 additions, 0 deletionsresearch/cv/dcgan/gpu_infer/src/utils.cc
- research/cv/dcgan/requirements.txt 3 additions, 0 deletionsresearch/cv/dcgan/requirements.txt
- research/cv/dcgan/scripts/run_distribute_train_ascend.sh 2 additions, 2 deletionsresearch/cv/dcgan/scripts/run_distribute_train_ascend.sh
- research/cv/dcgan/scripts/run_distribute_train_gpu.sh 75 additions, 0 deletionsresearch/cv/dcgan/scripts/run_distribute_train_gpu.sh
- research/cv/dcgan/scripts/run_eval_ascend.sh 66 additions, 0 deletionsresearch/cv/dcgan/scripts/run_eval_ascend.sh
- research/cv/dcgan/scripts/run_eval_gpu.sh 7 additions, 8 deletionsresearch/cv/dcgan/scripts/run_eval_gpu.sh
- research/cv/dcgan/scripts/run_infer_310.sh 2 additions, 2 deletionsresearch/cv/dcgan/scripts/run_infer_310.sh
- research/cv/dcgan/scripts/run_infer_gpu.sh 110 additions, 0 deletionsresearch/cv/dcgan/scripts/run_infer_gpu.sh
- research/cv/dcgan/scripts/run_standalone_train_ascend.sh 2 additions, 2 deletionsresearch/cv/dcgan/scripts/run_standalone_train_ascend.sh
- research/cv/dcgan/scripts/run_standalone_train_gpu.sh 70 additions, 0 deletionsresearch/cv/dcgan/scripts/run_standalone_train_gpu.sh
- research/cv/dcgan/train.py 36 additions, 18 deletionsresearch/cv/dcgan/train.py
- research/cv/dcgan/verify.py 1 addition, 14 deletionsresearch/cv/dcgan/verify.py
Please register or sign in to comment