Add proto in GPU.
Showing
- research/cv/ProtoNet/README.md 30 additions, 18 deletionsresearch/cv/ProtoNet/README.md
- research/cv/ProtoNet/scripts/run_distribution_gpu.sh 48 additions, 0 deletionsresearch/cv/ProtoNet/scripts/run_distribution_gpu.sh
- research/cv/ProtoNet/scripts/run_standalone_eval_gpu.sh 49 additions, 0 deletionsresearch/cv/ProtoNet/scripts/run_standalone_eval_gpu.sh
- research/cv/ProtoNet/scripts/run_standalone_train_gpu.sh 46 additions, 0 deletionsresearch/cv/ProtoNet/scripts/run_standalone_train_gpu.sh
- research/cv/ProtoNet/src/EvalCallBack.py 4 additions, 3 deletionsresearch/cv/ProtoNet/src/EvalCallBack.py
- research/cv/ProtoNet/train.py 30 additions, 13 deletionsresearch/cv/ProtoNet/train.py
Please register or sign in to comment