Cody Wong 93dd54136d [Fix] make arguments effective in tools/confusion_matrix.py (#1401) ...

* add an argument for customizing `title' of the output figure

* fix `color_theme' arguments not passing to plot function

Signed-off-by: code14 <mob5566@gmail.com>

2022-05-05 22:09:25 +08:00

..

convert_datasets

[Enhancement] Revise pre-commit-hooks (#1315)

2022-05-05 22:09:25 +08:00

model_converters

[Enhancement] Revise pre-commit-hooks (#1315)

2022-05-05 22:09:25 +08:00

optimizer_scripts

style: fix format so pep8 is satisfied

2022-04-12 14:26:54 +08:00

torchserve

[Enhancement] Revise pre-commit-hooks (#1315)

2022-05-05 22:09:25 +08:00

analyze_logs.py

[Dcos] Add header for files (#796)

2021-08-16 23:16:55 -07:00

benchmark.py

[Feature] benchmark can add work_dir and repeat times, test.py now has default work-dir (#1126)

2022-01-18 16:38:31 +08:00

browse_dataset.py

[Enhancement] Revise pre-commit-hooks (#1315)

2022-05-05 22:09:25 +08:00

confusion_matrix.py

[Fix] make arguments effective in tools/confusion_matrix.py (#1401)

2022-05-05 22:09:25 +08:00

deploy_test_kneron.py

style: fix format so pep8 is satisfied

2022-04-12 14:26:54 +08:00

deploy_test.py

Add deprecation message for deploy tool (#1321)

2022-05-05 22:09:25 +08:00

dist_test.sh

[Feature] Add multi machine dist_train. (#1383)

2022-05-05 22:09:25 +08:00

dist_train.sh

[Feature] Add multi machine dist_train. (#1383)

2022-05-05 22:09:25 +08:00

get_flops.py

Update get_flops.py (#1371)

2022-05-05 22:09:25 +08:00

onnx2tensorrt.py

Add deprecation message for deploy tool (#1321)

2022-05-05 22:09:25 +08:00

print_config.py

[Fix] revise --option to --options #1140

2021-12-15 15:24:08 +08:00

publish_model.py

[Dcos] Add header for files (#796)

2021-08-16 23:16:55 -07:00

pytorch2onnx_kneron.py

style: pytorch2onnx_kneron.py line too long

2022-04-11 15:53:36 +08:00

pytorch2onnx.py

Add deprecation message for deploy tool (#1321)

2022-05-05 22:09:25 +08:00

pytorch2torchscript.py

[Enhancement] Add codespell pre-commit hook and fix typos (#920)

2021-10-13 06:21:17 -07:00

slurm_test.sh

init commit

2020-07-10 02:39:01 +08:00

slurm_train.sh

init commit

2020-07-10 02:39:01 +08:00

test.py

[Enhance] New-style CPU training and inference. (#1251)

2022-01-28 23:03:01 +08:00

train.py

[Feature]: Add diff seeds to diff ranks and set torch seed in worker_init_fn (#1362)

2022-05-05 22:09:25 +08:00