github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-08-14 22:23:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

remove print

Browse Source
This commit is contained in:
Tong Li 2025-04-30 13:05:32 +08:00
parent e8cc51066b
commit e164d87fa5
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
applications/ColossalChat/coati/distributed/producer.py
View File

@ -39,7 +39,6 @@ class BaseProducer:
self.microbatch_size = microbatch_size self.microbatch_size = microbatch_size
assert batch_size % microbatch_size == 0 assert batch_size % microbatch_size == 0
self.num_microbatches = batch_size // microbatch_size self.num_microbatches = batch_size // microbatch_size
print("Num micro batches: ", self.num_microbatches)
self.dataset_config = dataset_config self.dataset_config = dataset_config
self.model_config = model_config self.model_config = model_config