Remove debug statement

This commit is contained in:
Daniel Povey 2022-11-09 14:18:23 +08:00
parent 3ff3f440ee
commit d1d4be8ecc

View File

@ -1009,9 +1009,6 @@ def run(rank, world_size, args):
num_param = sum([p.numel() for p in model.parameters()])
logging.info(f"Number of model parameters: {num_param}")
for name, module in model.named_modules():
num_param = sum([p.numel() for p in module.parameters()])
logging.info(f"Number of model parameters for {name}: {num_param}")
assert params.save_every_n >= params.average_period
model_avg: Optional[nn.Module] = None