InternLM/internlm
Sun Peng 860de0aa46
Feat/add runntime gpu test (#254)
* feat: add gpu bench

* feat/add allreduce runtime bench

---------

Co-authored-by: sunpengsdu <sunpengsdu@gmail.com>
2023-09-01 13:38:01 +08:00
..
apis initial commit 2023-07-06 12:55:23 +08:00
core Feat/add runntime gpu test (#254) 2023-09-01 13:38:01 +08:00
data Merge develop to main (#233) 2023-08-24 22:03:04 +08:00
initialize feat(model): implement uniform_init for tensor. (#252) 2023-09-01 01:12:53 +08:00
model fix(metric): argument missing in getting loss metrics. (#256) 2023-08-31 17:44:39 +08:00
monitor Merge develop to main (#233) 2023-08-24 22:03:04 +08:00
solver feat(utils/writer.py): support writer add_scalars for writing dict data (#257) 2023-09-01 13:24:46 +08:00
train feat(utils/writer.py): support writer add_scalars for writing dict data (#257) 2023-09-01 13:24:46 +08:00
utils Feat/add runntime gpu test (#254) 2023-09-01 13:38:01 +08:00
__init__.py initial commit 2023-07-06 12:55:23 +08:00