Yan Bai
commited on
Commit
·
7e32279
1
Parent(s):
85d63c1
reduce printed info
Browse files- estimate_013.py +1 -1
estimate_013.py
CHANGED
|
@@ -352,7 +352,7 @@ def report_memory_usage_one_pp_rank(
|
|
| 352 |
)
|
| 353 |
|
| 354 |
one_chunk.__repr__()
|
| 355 |
-
print(one_chunk)
|
| 356 |
print(
|
| 357 |
f"Number of parameters in every GPU in billions: "
|
| 358 |
f"{num_parameter_this_shard / 10**9: .2f} where mlp part is {num_parameter_this_shard_sparse / 10**9: .2f}"
|
|
|
|
| 352 |
)
|
| 353 |
|
| 354 |
one_chunk.__repr__()
|
| 355 |
+
# print(one_chunk)
|
| 356 |
print(
|
| 357 |
f"Number of parameters in every GPU in billions: "
|
| 358 |
f"{num_parameter_this_shard / 10**9: .2f} where mlp part is {num_parameter_this_shard_sparse / 10**9: .2f}"
|