mmpretrain/tools
LXXXXR e76c5a368d
[Feature] Support fp16 training (#178)
* change mmcls fp16 to mmcv hook

* support fp16

* clean unnessary stuff
2021-03-17 15:53:55 +08:00
..
convert_models Add VGG and pretained models (#27) 2020-09-29 17:49:42 +08:00
analyze_results.py [Feature] Add analyze_results.py to select imgs for success/fail (#142) 2021-01-19 15:49:14 +08:00
dist_test.sh init commit 2020-05-21 21:21:43 +08:00
dist_train.sh init commit 2020-05-21 21:21:43 +08:00
get_flops.py Add two useful tools from mmdet 2020-07-12 00:06:53 +08:00
onnx2tensorrt.py add onnx2tensorrt (#153) 2021-02-05 17:48:37 +08:00
publish_model.py Add two useful tools from mmdet 2020-07-12 00:06:53 +08:00
pytorch2onnx.py support dynamic shape export to onnx (#175) 2021-03-17 15:50:03 +08:00
slurm_test.sh init commit 2020-05-21 21:21:43 +08:00
slurm_train.sh init commit 2020-05-21 21:21:43 +08:00
test.py [Feature] Support fp16 training (#178) 2021-03-17 15:53:55 +08:00
train.py Fix bug in gpu_ids in distributed training (#107) 2020-12-09 16:28:03 +08:00