from local

This commit is contained in:
dohe0342 2023-01-08 17:42:54 +09:00
parent af71e98e27
commit d5504a8d7e
3 changed files with 3 additions and 0 deletions

Binary file not shown.

View File

@ -966,6 +966,9 @@ def run(rank, world_size, args):
logging.info("About to create model")
model = get_transducer_model(params)
print(model)
exit()
num_param = sum([p.numel() for p in model.parameters()])
logging.info(f"Number of model parameters: {num_param}")