wangliujun 912d109dbe Update get_flops.py (#1371) ...

* Update get_flops.py

argparse.ArgumentParser(description='Get the FLOPs of a segmentor')

* Update get_flops.py

2022-03-14 21:02:43 +08:00

..

convert_datasets

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

2022-02-23 23:44:27 +08:00

model_converters

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

2022-02-23 23:44:27 +08:00

torchserve

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

2022-02-23 23:44:27 +08:00

analyze_logs.py

…

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-02-23 23:44:27 +08:00

confusion_matrix.py

[Feature] Generating and plotting confusion matrix (#1301)

2022-03-03 16:34:33 +08:00

deploy_test.py

Add deprecation message for deploy tool (#1321)

2022-02-25 11:22:03 +08:00

dist_test.sh

…

dist_train.sh

…

get_flops.py

Update get_flops.py (#1371)

2022-03-14 21:02:43 +08:00

onnx2tensorrt.py

Add deprecation message for deploy tool (#1321)

2022-02-25 11:22:03 +08:00

print_config.py

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

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

publish_model.py

…

pytorch2onnx.py

Add deprecation message for deploy tool (#1321)

2022-02-25 11:22:03 +08:00

pytorch2torchscript.py

…

slurm_test.sh

…

slurm_train.sh

…

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-03-10 09:41:16 +08:00