from local

This commit is contained in:
dohe0342 2023-01-09 20:19:54 +09:00
parent 17328f6348
commit 8b59cb1ac0
3 changed files with 0 additions and 2 deletions

View File

@ -982,7 +982,6 @@ def run(rank, world_size, args):
transducer_model.load_state_dict(pre_trained_model, strict=True)
model = get_interformer_model(transducer_model.encoder, params)
print(model)
'''
for n, p in model.named_parameters():
if 'layer' not in n:
@ -998,7 +997,6 @@ def run(rank, world_size, args):
else:
print(f'skipping param load {n}')
'''
exit()
num_param = sum([p.numel() for p in model.parameters()])
logging.info(f"Number of model parameters: {num_param}")