From 9ee285cf9533703d9d8ba51f04b6df58e97fab6f Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Micha=C5=82=20Koz=C5=82owski?= Date: Sat, 7 Jan 2023 14:41:26 +0100 Subject: [PATCH] fix --- image_download.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/image_download.py b/image_download.py index 2d5f0b4..3e68351 100644 --- a/image_download.py +++ b/image_download.py @@ -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"]