From 689dd6675dd98ddf6d093f318e04610426ead64d Mon Sep 17 00:00:00 2001 From: ChairC <974833488@qq.com> Date: Fri, 31 May 2024 12:40:08 +0800 Subject: [PATCH] Update: Modify type description error in README. --- README.md | 4 ++-- README_zh.md | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/README.md b/README.md index afcf4ba..7a0a4eb 100644 --- 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 index c8528cf..0b2dd11 100644 --- 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的编号 |