From d001fdf750fc47327fdcbb8f41508b5334edebc6 Mon Sep 17 00:00:00 2001 From: Chengqi Deng Date: Thu, 28 Apr 2022 14:55:41 +0800 Subject: [PATCH] Fix argparse name --- train_gpt.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train_gpt.py b/train_gpt.py index 2b069ed..f7df125 100644 --- a/train_gpt.py +++ b/train_gpt.py @@ -23,7 +23,7 @@ # CONFIG ########################################### -parser = argparse.ArgumentParser(description="Train VQGAN") +parser = argparse.ArgumentParser(description="Train GPT") parser.add_argument("--ds", type=str, default="coco", help="dataset name") parser.add_argument("--gpt", type=str, default="gpt2_medium", help="GPT model") parser.add_argument("--dropout", type=float, default=0.1, help="dropout rate")