mirror of
https://github.com/storytold/FineTrainers-Conditioning.git
synced 2026-10-18 07:33:20 +00:00
fix: correct type in .py files (#52)
This commit is contained in:
@@ -599,7 +599,7 @@ def main(args):
|
|||||||
if args.resume_from_checkpoint != "latest":
|
if args.resume_from_checkpoint != "latest":
|
||||||
path = os.path.basename(args.resume_from_checkpoint)
|
path = os.path.basename(args.resume_from_checkpoint)
|
||||||
else:
|
else:
|
||||||
# Get the mos recent checkpoint
|
# Get the most recent checkpoint
|
||||||
dirs = os.listdir(args.output_dir)
|
dirs = os.listdir(args.output_dir)
|
||||||
dirs = [d for d in dirs if d.startswith("checkpoint")]
|
dirs = [d for d in dirs if d.startswith("checkpoint")]
|
||||||
dirs = sorted(dirs, key=lambda x: int(x.split("-")[1]))
|
dirs = sorted(dirs, key=lambda x: int(x.split("-")[1]))
|
||||||
|
|||||||
@@ -583,7 +583,7 @@ def main(args):
|
|||||||
if args.resume_from_checkpoint != "latest":
|
if args.resume_from_checkpoint != "latest":
|
||||||
path = os.path.basename(args.resume_from_checkpoint)
|
path = os.path.basename(args.resume_from_checkpoint)
|
||||||
else:
|
else:
|
||||||
# Get the mos recent checkpoint
|
# Get the most recent checkpoint
|
||||||
dirs = os.listdir(args.output_dir)
|
dirs = os.listdir(args.output_dir)
|
||||||
dirs = [d for d in dirs if d.startswith("checkpoint")]
|
dirs = [d for d in dirs if d.startswith("checkpoint")]
|
||||||
dirs = sorted(dirs, key=lambda x: int(x.split("-")[1]))
|
dirs = sorted(dirs, key=lambda x: int(x.split("-")[1]))
|
||||||
|
|||||||
@@ -547,7 +547,7 @@ def main(args):
|
|||||||
if args.resume_from_checkpoint != "latest":
|
if args.resume_from_checkpoint != "latest":
|
||||||
path = os.path.basename(args.resume_from_checkpoint)
|
path = os.path.basename(args.resume_from_checkpoint)
|
||||||
else:
|
else:
|
||||||
# Get the mos recent checkpoint
|
# Get the most recent checkpoint
|
||||||
dirs = os.listdir(args.output_dir)
|
dirs = os.listdir(args.output_dir)
|
||||||
dirs = [d for d in dirs if d.startswith("checkpoint")]
|
dirs = [d for d in dirs if d.startswith("checkpoint")]
|
||||||
dirs = sorted(dirs, key=lambda x: int(x.split("-")[1]))
|
dirs = sorted(dirs, key=lambda x: int(x.split("-")[1]))
|
||||||
|
|||||||
Reference in New Issue
Block a user