Update: Modify type description error in README.

chairc · May 31, 2024 · 689dd66 · 689dd66
1 parent 0bb876e
commit 689dd66
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/README.md b/README.md
@@ -453,8 +453,8 @@ Of course, based on the 64×64 U-Net model, we generate 160×160 `NEU-DET` image
 | path               | Path                        |      str       | Input two paths: the generated set path and the training set path in evaluation mode; input path and output path in npz mode |
 | --batch_size       | Training batch size         |      int       | Size of each training batch                                  |
 | --num_workers      | Number of loading processes |      int       | Number of subprocesses used for data loading. It consumes a large amount of CPU and memory but can speed up training |
-| --dims             | Dimensions                  |      str       | The dimensions of the Inception features to use              |
-| --save_stats       | Save stats                  |      str       | Generate npz archives from the sample directory              |
+| --dims             | Dimensions                  |      int       | The dimensions of the Inception features to use              |
+| --save_stats       | Save stats                  |      bool      | Generate npz archives from the sample directory              |
 | --use_gpu          | Specify GPU                 |      int       | Generally used to set the specific GPU for training, input the GPU number |
 
 ### About Citation

diff --git a/README_zh.md b/README_zh.md
@@ -452,8 +452,8 @@ Integrated Design Diffusion Model
 | path          | 路径              |   str    | 输入2个路径，评估模式下为生成集路径和训练集路径；npz模式下为输入路径和输出路径 |
 | --batch_size  | 训练批次          |   int    | 训练批次大小                                                 |
 | --num_workers | 加载进程数量      |   int    | 用于数据加载的子进程数量，大量占用CPU和内存，但可以加快训练速度 |
-| --dims        | 维度              |   str    | 要使用的 Inception 功能的维度                                |
-| --save_stats  | 保存存档          |   str    | 从样本目录生成 npz 存档                                      |
+| --dims        | 维度              |   int    | 要使用的 Inception 功能的维度                                |
+| --save_stats  | 保存存档          |   bool   | 从样本目录生成 npz 存档                                      |
 | --use_gpu     | 设置运行指定的GPU |   int    | 一般训练中设置指定的运行GPU，输入为GPU的编号                 |