Exit if failed to load model

richardboo · May 16, 2016 · ceca9d8 · ceca9d8
1 parent 5f657da
commit ceca9d8
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/examples/imagenet/validate.py b/examples/imagenet/validate.py
@@ -77,6 +77,8 @@ def main():
 
     # Load the network
     net = load_model(args.model)
+    if net is None:
+        exit(-1)
 
     # Load the dataset
     data_spec = models.get_data_spec(model_instance=net)