LXXXXR f355f15485 Fix bug in gpu_ids in distributed training (#107) ... * update gpu_ids in distributed training * move linear scaling rule after getting correct gpu_ids * Remove support for autoscale_lr		2020-12-09 16:28:03 +08:00
..
convert_models	Add VGG and pretained models (#27)	2020-09-29 17:49:42 +08:00
dist_test.sh	…
dist_train.sh	…
get_flops.py	…
publish_model.py	…
pytorch2onnx.py	Add pytorch2onnx (#20)	2020-09-30 20:13:36 +08:00
slurm_test.sh	…
slurm_train.sh	…
test.py	add argument to override configs (#91)	2020-11-20 13:50:28 +08:00
train.py	Fix bug in gpu_ids in distributed training (#107)	2020-12-09 16:28:03 +08:00