Update train_depth.py

jeffsonyu · Jul 20, 2023 · a4c169b · a4c169b
1 parent 3d79f57
commit a4c169b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train_depth.py b/train_depth.py
@@ -135,7 +135,7 @@
 checkpoint_io = CheckpointIO(out_dir, model=model, optimizer=optimizer)
 try:
     load_dict = checkpoint_io.load(cfg['test']['model_file'], device=device)
-except FileExistsError:
+except FileNotFoundError:
     load_dict = dict()
 
 epoch_it = load_dict.get('epoch_it', 0)