!310 add simclr gpu
Merge pull request !310 from MapleGrove/master
No related branches found
No related tags found
Showing
- official/cv/simclr/README.md 48 additions, 34 deletionsofficial/cv/simclr/README.md
- official/cv/simclr/linear_eval.py 5 additions, 5 deletionsofficial/cv/simclr/linear_eval.py
- official/cv/simclr/scripts/run_distribution_train_ascend.sh 1 addition, 1 deletionofficial/cv/simclr/scripts/run_distribution_train_ascend.sh
- official/cv/simclr/scripts/run_distribution_train_gpu.sh 53 additions, 0 deletionsofficial/cv/simclr/scripts/run_distribution_train_gpu.sh
- official/cv/simclr/scripts/run_standalone_eval_cpu.sh 37 additions, 0 deletionsofficial/cv/simclr/scripts/run_standalone_eval_cpu.sh
- official/cv/simclr/scripts/run_standalone_eval_gpu.sh 38 additions, 0 deletionsofficial/cv/simclr/scripts/run_standalone_eval_gpu.sh
- official/cv/simclr/scripts/run_standalone_train_gpu.sh 32 additions, 0 deletionsofficial/cv/simclr/scripts/run_standalone_train_gpu.sh
- official/cv/simclr/src/dataset.py 7 additions, 3 deletionsofficial/cv/simclr/src/dataset.py
- official/cv/simclr/src/optimizer.py 11 additions, 5 deletionsofficial/cv/simclr/src/optimizer.py
- official/cv/simclr/train.py 11 additions, 3 deletionsofficial/cv/simclr/train.py
Please register or sign in to comment