From 162b99b4a32eded2ece10c60794e57f560df22ae Mon Sep 17 00:00:00 2001 From: Srini Iyer Date: Thu, 6 Feb 2025 00:26:37 +0000 Subject: [PATCH] Log model --- bytelatent/train.py | 1 + 1 file changed, 1 insertion(+) diff --git a/bytelatent/train.py b/bytelatent/train.py index 86d1c7a..bb8307a 100644 --- a/bytelatent/train.py +++ b/bytelatent/train.py @@ -325,6 +325,7 @@ def train(args: TrainArgs): # log model size + logger.info(model) logger.info(f"Model size: {model_param_count:,} total parameters") gpu_memory_monitor = GPUMemoryMonitor("cuda")