diff --git a/example/image-classification/train_imagenet.py b/example/image-classification/train_imagenet.py index 5760a9af3782..f465fbc5f469 100644 --- 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)