This commit is contained in:
Michał Kozłowski 2023-01-07 14:41:26 +01:00
parent 0f3a6eb9da
commit 9ee285cf95

View File

@ -23,7 +23,7 @@ def main(args):
print(f"Creating missing folder: {args.output_folder}")
os.mkdir(args.output_folder)
if args.from_checkpoint:
if args.from_checkpoint and os.path.exists("./state.pickle"):
with open("state.pickle", "rb") as state:
state_dict = pickle.load(state)
offset = state_dict["row_index"]