minor

alibaba · Jan 29, 2024 · 2691b0b · 2691b0b
1 parent 08decba
commit 2691b0b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/examples/igbh/dist_train_rgnn.py b/examples/igbh/dist_train_rgnn.py
@@ -189,7 +189,8 @@ def run_training_proc(local_proc_rank, num_nodes, node_rank, num_training_procs,
   if ckpt_path is not None:
     try:
       ckpt = torch.load(ckpt_path)
-    except FileNotFoundError:
+    except FileNotFoundError as e:
+      print(f"Checkpoint file not found: {e}")
       return -1
 
   # Define model and optimizer.

diff --git a/examples/igbh/train_rgnn_multi_gpu.py b/examples/igbh/train_rgnn_multi_gpu.py
@@ -124,7 +124,8 @@ def run_training_proc(rank, world_size,
   if ckpt_path is not None:
     try:
       ckpt = torch.load(ckpt_path)
-    except FileNotFoundError:
+    except FileNotFoundError as e:
+      print(f"Checkpoint file not found: {e}")
       return -1
 
   # Define model and optimizer.