From fcc6706adc36f1ae4a9345e79acc28e76acea918 Mon Sep 17 00:00:00 2001 From: Ragul Balaji Date: Mon, 2 Sep 2024 18:15:11 +0800 Subject: [PATCH] Update README.md --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index e2b96d3..0ae5e74 100644 --- a/README.md +++ b/README.md @@ -80,7 +80,7 @@ python train.py ``` **Options** -* `TRAIN_IMAGE_SIZE`: sets the dimension (height and weight) of training images. Bigger GPU memory is needed to train with larger images. Default is `256`px. +* `TRAIN_IMAGE_SIZE`: sets the dimension (height and width) of training images. Bigger GPU memory is needed to train with larger images. Default is `256`px. * `DATASET_PATH`: folder containing the MS-COCO `train2014` images. Default is `"dataset"` * `NUM_EPOCHS`: Number of epochs of training pass. Default is `1` with 40,000 training images * `STYLE_IMAGE_PATH`: path of the style image