from local

This commit is contained in:
dohe0342 2023-01-10 00:59:53 +09:00
parent e310b6f783
commit 86df081ca0
2 changed files with 1 additions and 2 deletions

View File

@ -973,8 +973,7 @@ def run(rank, world_size, args):
# if 'layer' not in n:
for n, p in pre_trained_model.items():
print(n)
exit()
num_param = sum([p.numel() for p in model.parameters()])
logging.info(f"Number of model parameters: {num_param}")