Merge pull request #80 from chairc/dev

Update: Update the README.

README.md

@@ -46,6 +46,7 @@ If you have any questions, please check [**the existing issues**](https://github
 Integrated Design Diffusion Model
 ├── config
 │   ├── choices.py
+│   ├── setting.py
 │   └── version.py
 ├── datasets
 │   └── dataset_demo
@@ -275,6 +276,7 @@ The training GPU implements environment for this README is as follows: models ar
 | --dataset_path         |             | Dataset path                    | str  | Path to the conditional dataset, such as CIFAR-10, with each class in a separate folder, or the path to the unconditional dataset with all images in one folder |
 | --amp                 |             | Automatic mixed precision training | bool | Enable automatic mixed precision training. It effectively reduces GPU memory usage but may affect training accuracy and results |
 | --optim                |             | Optimizer                       | str  | Optimizer selection. Currently supports Adam and AdamW       |
+| --loss                 |             | Loss function                   | str  | Loss selection. Currently supports MSELoss, L1Loss, HuberLoss and SmoothL1Loss             |
 | --act | | Activation function | str | Activation function selection. Currently supports gelu, silu, relu, relu6 and lrelu |
 | --lr                   |             | Learning rate                   | float | Initial learning rate. |
 | --lr_func | | Learning rate schedule | str | Setting learning rate schedule, currently supporting linear, cosine, and warmup_cosine. |

README_zh.md

@@ -45,6 +45,7 @@
 Integrated Design Diffusion Model
 ├── config
 │   ├── choices.py
+│   ├── setting.py
 │   └── version.py
 ├── datasets
 │   └── dataset_demo
@@ -274,6 +275,7 @@ Integrated Design Diffusion Model
 | --dataset_path         |          | 数据集路径                       |   str    | 有条件数据集，例如cifar10，每个类别一个文件夹，路径为主文件夹；无条件数据集，所有图放在一个文件夹，路径为图片文件夹 |
 | --amp                  |          | 混合精度训练                     |   bool   | 开启混合精度训练，有效减少显存使用，但无法保证训练精度和训练结果 |
 | --optim                |          | 优化器                           |   str    | 优化器选择，目前支持adam和adamw                              |
+| --loss                 |          | 损失函数                         |   str    | 损失函数选择，目前支持MSELoss、L1Loss、HuberLoss和moothL1Loss |
 | --act                  |          | 激活函数                         |   str    | 激活函数选择，目前支持gelu、silu、relu、relu6和lrelu         |
 | --lr                   |          | 学习率                           |  float   | 初始化学习率                                                 |
 | --lr_func              |          | 学习率方法                       |   str    | 设置学习率方法，当前支持linear、cosine和warmup_cosine        |