ColossalAI/colossalai/cli/benchmark
アマデウス 49715a78f0 [NFC] polish colossalai/cli/benchmark/benchmark.py code style (#2287) 2023-01-04 15:09:57 +08:00
..
__init__.py [cli] remove unused imports (#1001) 2022-05-18 23:27:18 +08:00
benchmark.py [NFC] polish colossalai/cli/benchmark/benchmark.py code style (#2287) 2023-01-04 15:09:57 +08:00
models.py [cli] refactored micro-benchmarking cli and added more metrics (#858) 2022-04-25 11:48:07 +08:00
utils.py [hotfix] fix some bugs caused by size mismatch. (#1011) 2022-05-23 14:02:28 +08:00