GitHub - bhwangfy/Imagenet_training: ImageNet Repo Used to Compare Different Optimizers

bhwangfy / Imagenet_training Public

ImageNet Repo Used to Compare Different Optimizers

Name		Name	Last commit message	Last commit date
Latest commit History 2 Commits
.ipynb_checkpoints		.ipynb_checkpoints
Plot		Plot
__pycache__		__pycache__
utils		utils
.DS_Store		.DS_Store
README.txt		README.txt
amlt_search_imagenet.yml		amlt_search_imagenet.yml
models.py		models.py
optimizers.py		optimizers.py
plot.py		plot.py
requirements.txt		requirements.txt
run.sh		run.sh
toy.log		toy.log
trainer.py		trainer.py

Repository files navigation

## 多GPU启动指令

使用"torchrun --nproc_per_node=2 trainer.py"启动

  - nproc_per_node = 并行GPU数量
  - 使用“CUDA_VISIBLE_DEVICES=x,y“指定GPU