mmclassification/tools
Lei Yang edffbcc486
Add pytorch2onnx (#20)
* add pytorch2onnx

* add pytorch2onnx API
2020-09-30 20:13:36 +08:00
..
convert_models Add VGG and pretained models (#27) 2020-09-29 17:49:42 +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
publish_model.py Add two useful tools from mmdet 2020-07-12 00:06:53 +08:00
pytorch2onnx.py Add pytorch2onnx (#20) 2020-09-30 20:13:36 +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 Add model inference (#16) 2020-09-30 19:00:20 +08:00
train.py Add classifiers, heads, necks and losses 2020-07-07 19:32:06 +08:00