!2503 Implemented training and evaluation of metric_learn on a GPU
Merge pull request !2503 from Vladimir Nechaev/metric_learn_gpu
No related branches found
No related tags found
Showing
- research/cv/metric_learn/README.md 498 additions, 0 deletionsresearch/cv/metric_learn/README.md
- research/cv/metric_learn/eval.py 5 additions, 2 deletionsresearch/cv/metric_learn/eval.py
- research/cv/metric_learn/requirements.txt 1 addition, 0 deletionsresearch/cv/metric_learn/requirements.txt
- research/cv/metric_learn/scripts/run_distribute_train_gpu.sh 78 additions, 0 deletionsresearch/cv/metric_learn/scripts/run_distribute_train_gpu.sh
- research/cv/metric_learn/scripts/run_eval_gpu.sh 72 additions, 0 deletionsresearch/cv/metric_learn/scripts/run_eval_gpu.sh
- research/cv/metric_learn/scripts/run_standalone_train_gpu.sh 73 additions, 0 deletionsresearch/cv/metric_learn/scripts/run_standalone_train_gpu.sh
- research/cv/metric_learn/src/config_gpu.py 78 additions, 0 deletionsresearch/cv/metric_learn/src/config_gpu.py
- research/cv/metric_learn/src/dataset.py 2 additions, 0 deletionsresearch/cv/metric_learn/src/dataset.py
- research/cv/metric_learn/src/dataset_gpu.py 76 additions, 0 deletionsresearch/cv/metric_learn/src/dataset_gpu.py
- research/cv/metric_learn/src/loss.py 7 additions, 2 deletionsresearch/cv/metric_learn/src/loss.py
- research/cv/metric_learn/src/resnet.py 1 addition, 1 deletionresearch/cv/metric_learn/src/resnet.py
- research/cv/metric_learn/src/utility.py 13 additions, 0 deletionsresearch/cv/metric_learn/src/utility.py
- research/cv/metric_learn/train.py 148 additions, 49 deletionsresearch/cv/metric_learn/train.py
Please register or sign in to comment