Fix description of argument parser (apache#7507)

reminisce · Aug 17, 2017 · 462dee7 · 462dee7
1 parent 1286809
commit 462dee7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/example/image-classification/train_imagenet.py b/example/image-classification/train_imagenet.py
@@ -25,7 +25,7 @@
 
 if __name__ == '__main__':
     # parse args
-    parser = argparse.ArgumentParser(description="train cifar10",
+    parser = argparse.ArgumentParser(description="train imagenet-1k",
                                      formatter_class=argparse.ArgumentDefaultsHelpFormatter)
     fit.add_fit_args(parser)
     data.add_data_args(parser)