Fix argparse name

HFAiLab · Apr 28, 2022 · d001fdf · d001fdf
1 parent e8d289e
commit d001fdf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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")