Exit if failed to load model

Saumitro Dasgupta · Saumitro Dasgupta · commit ceca9d83fe45 · 2016-05-16T00:47:53.000-07:00
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)
