from local
This commit is contained in:
parent
e310b6f783
commit
86df081ca0
Binary file not shown.
@ -973,8 +973,7 @@ def run(rank, world_size, args):
|
|||||||
# if 'layer' not in n:
|
# if 'layer' not in n:
|
||||||
for n, p in pre_trained_model.items():
|
for n, p in pre_trained_model.items():
|
||||||
print(n)
|
print(n)
|
||||||
|
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}")
|
||||||
|
|||||||
Reference in New Issue
Block a user