fix
This commit is contained in:
parent
0f3a6eb9da
commit
9ee285cf95
@ -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"]
|
||||
|
Loading…
Reference in New Issue
Block a user