from local
This commit is contained in:
parent
3638abf352
commit
fb79717755
Binary file not shown.
@ -1259,7 +1259,6 @@ def run(rank, world_size, args, wb=None):
|
|||||||
logging.info("About to create model")
|
logging.info("About to create model")
|
||||||
model = get_transducer_model(params)
|
model = get_transducer_model(params)
|
||||||
logging.info(model)
|
logging.info(model)
|
||||||
exit()
|
|
||||||
|
|
||||||
num_param = sum([p.numel() for p in model.parameters()])
|
num_param = sum([p.numel() for p in model.parameters()])
|
||||||
logging.info(f"Number of model parameters: {num_param}")
|
logging.info(f"Number of model parameters: {num_param}")
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user