vikarti.anatra/blt
1
0
Fork 0
mirror of https://github.com/facebookresearch/blt.git synced 2025-02-23 05:22:16 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Log model

Browse source
This commit is contained in:
Srini Iyer 2025-02-06 00:26:37 +00:00
parent 7cf8fab49b
commit 162b99b4a3
1 changed files with 1 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
bytelatent/train.py
View file

@ -325,6 +325,7 @@ def train(args: TrainArgs):
# log model size # log model size
logger.info(model)
logger.info(f"Model size: {model_param_count:,} total parameters") logger.info(f"Model size: {model_param_count:,} total parameters")
gpu_memory_monitor = GPUMemoryMonitor("cuda") gpu_memory_monitor = GPUMemoryMonitor("cuda")