diff --git a/vision_transformer/README.md b/vision_transformer/README.md index 7a4cd1cc94..4ce26beb80 100644 --- a/vision_transformer/README.md +++ b/vision_transformer/README.md @@ -27,7 +27,7 @@ options: --num-encoders NUM_ENCODERS number of encoders (default : 12) --dropout DROPOUT dropout value (default : 0.1) - --img-size IMG_SIZE image size to be reshaped to (default : 224 + --img-size IMG_SIZE image size to be reshaped to (default : 224) --num-classes NUM_CLASSES number of classes in dataset (default : 10 for CIFAR10) --epochs EPOCHS number of epochs (default : 10) @@ -37,4 +37,4 @@ options: --batch-size BATCH_SIZE batch size (default : 4) -``` \ No newline at end of file +```