from local

This commit is contained in:
dohe0342 2023-01-10 01:19:14 +09:00
parent 1d1d455593
commit 4d7559e53b
2 changed files with 3 additions and 4 deletions

View File

@ -974,10 +974,9 @@ def run(rank, world_size, args):
pre_trained2 = torch.load(path2)
pre_trained_model2 = pre_trained2["model"]
for n, p in pre_trained_model2.items():
if rank == 0:
print(n)
exit()
#for n, p in pre_trained_model2.items():
# if rank == 0:
# print(n)
for n, p in model.named_parameters():
if 'layer' not in n: