from local

This commit is contained in:
dohe0342 2023-01-10 01:13:46 +09:00
parent b9ef1eb801
commit bf7f4305c1
2 changed files with 1 additions and 0 deletions

View File

@ -977,6 +977,7 @@ def run(rank, world_size, args):
for n, p in pre_trained_model2.items():
if rank == 0:
print(n)
exit()
for n, p in model.named_parameters():
if 'layer' not in n: