Merge pull request bowang-lab#276 from Zehui-Lin/patch-1

Fix typos
VersaceSu7 · Jun 27, 2024 · a7b7776 · a7b7776
2 parents 3ead5b7 + 684a217
commit a7b7776
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train_one_gpu.py b/train_one_gpu.py
@@ -157,7 +157,7 @@ def __getitem__(self, index):
 )
 # parser.add_argument('-device', type=str, default='cuda:0')
 parser.add_argument(
-    "--load_pretrain", type=bool, default=True, help="use wandb to monitor training"
+    "--load_pretrain", type=bool, default=True, help="load pretrain model"
 )
 parser.add_argument("-pretrain_model_path", type=str, default="")
 parser.add_argument("-work_dir", type=str, default="./work_dir")