Fix args description (#1209)

pytorch · Jan 12, 2024 · 5921fc1 · 5921fc1
1 parent 3e56db2
commit 5921fc1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/vision_transformer/main.py b/vision_transformer/main.py
@@ -218,7 +218,7 @@ def main():
     parser.add_argument('--n-channels', type=int, default=3,
                         help='number of channels in images (default : 3 for RGB)')
     parser.add_argument('--num-heads', type=int, default=12,
-                        help='(default : 16)')
+                        help='(default : 12)')
     parser.add_argument('--num-encoders', type=int, default=12,
                         help='number of encoders (default : 12)')
     parser.add_argument('--dropout', type=int, default=0.1,