Browse Source

update doc

will-jl944 4 years ago
parent
commit
17acddd9a3

+ 18 - 18
docs/apis/models/classification.md

@@ -30,24 +30,24 @@ train(self, num_epochs, train_dataset, train_batch_size=64, eval_dataset=None, o
 >
 > **参数**
 >
-> > - **num_epochs** (int): 训练迭代轮数。
-> > - **train_dataset** (paddlex.dataset): 训练数据集。
-> > - **train_batch_size** (int): 训练数据batch大小。同时作为验证数据batch大小。默认为64。
-> > - **eval_dataset** (paddlex.dataset or None): 评估数据集。当该参数为None时,训练过程中不会进行模型评估。默认为None。
-> > - **optimizer** (paddle.optimizer.Optimizer): 优化器。当该参数为None时,使用默认优化器:paddle.optimizer.lr.PiecewiseDecay衰减策略,paddle.optimizer.Momentum优化方法。
-> > - **save_interval_epochs** (int): 模型保存间隔(单位:迭代轮数)。默认为1。
-> > - **log_interval_steps** (int): 训练日志输出间隔(单位:迭代步数)。默认为10。
-> > - **save_dir** (str): 模型保存路径。默认为'output'。
-> > - **pretrain_weights** (str or None): 若指定为'.pdparams'文件时,则从文件加载模型权重;若为字符串'IMAGENET',则自动下载在ImageNet图片数据上预训练的模型权重;若为None,则不使用预训练模型。默认为'IMAGENET'。
-> > - **learning_rate** (float): 默认优化器的初始学习率。默认为0.025。
-> > - **warmup_steps** (int): 默认优化器的warmup步数,学习率将在设定的步数内,从warmup_start_lr线性增长至设定的learning_rate,默认为0。
-> > - **warmup_start_lr**(float): 默认优化器的warmup起始学习率,默认为0.0。
-> > - **lr_decay_epochs** (list): 默认优化器的学习率衰减轮数。默认为[30, 60, 90]。
-> > - **lr_decay_gamma** (float): 默认优化器的学习率衰减率。默认为0.1。
-> > - **early_stop** (bool): 是否使用提前终止训练策略。默认为False。
-> > - **early_stop_patience** (int): 当使用提前终止训练策略时,如果验证集精度在`early_stop_patience`个epoch内连续下降或持平,则终止训练。默认为5。
-> > - **use_vdl** (bool): 是否使用VisualDL进行可视化。默认为True。
-> > - **resume_checkpoint** (str): 恢复训练时指定上次训练保存的模型路径,例如`output/mobilenetv3_small/best_model`。若为None,则不会恢复训练。默认值为None。
+- **num_epochs** (int): 训练迭代轮数。
+- **train_dataset** (paddlex.dataset): 训练数据集。
+- **train_batch_size** (int): 训练数据batch大小。同时作为验证数据batch大小。默认为64。
+- **eval_dataset** (paddlex.dataset or None): 评估数据集。当该参数为None时,训练过程中不会进行模型评估。默认为None。
+- **optimizer** (paddle.optimizer.Optimizer): 优化器。当该参数为None时,使用默认优化器:paddle.optimizer.lr.PiecewiseDecay衰减策略,paddle.optimizer.Momentum优化方法。
+- **save_interval_epochs** (int): 模型保存间隔(单位:迭代轮数)。默认为1。
+- **log_interval_steps** (int): 训练日志输出间隔(单位:迭代步数)。默认为10。
+- **save_dir** (str): 模型保存路径。默认为'output'。
+- **pretrain_weights** (str or None): 若指定为'.pdparams'文件时,则从文件加载模型权重;若为字符串'IMAGENET',则自动下载在ImageNet图片数据上预训练的模型权重;若为None,则不使用预训练模型。默认为'IMAGENET'。
+- **learning_rate** (float): 默认优化器的初始学习率。默认为0.025。
+- **warmup_steps** (int): 默认优化器的warmup步数,学习率将在设定的步数内,从warmup_start_lr线性增长至设定的learning_rate,默认为0。
+- **warmup_start_lr**(float): 默认优化器的warmup起始学习率,默认为0.0。
+- **lr_decay_epochs** (list): 默认优化器的学习率衰减轮数。默认为[30, 60, 90]。
+- **lr_decay_gamma** (float): 默认优化器的学习率衰减率。默认为0.1。
+- **early_stop** (bool): 是否使用提前终止训练策略。默认为False。
+- **early_stop_patience** (int): 当使用提前终止训练策略时,如果验证集精度在`early_stop_patience`个epoch内连续下降或持平,则终止训练。默认为5。
+- **use_vdl** (bool): 是否使用VisualDL进行可视化。默认为True。
+- **resume_checkpoint** (str): 恢复训练时指定上次训练保存的模型路径,例如`output/mobilenetv3_small/best_model`。若为None,则不会恢复训练。默认值为None。**恢复训练需要将`pretrain_weights`设置为None。**
 
 ### <h3 id="12">evaluate</h3>
 

+ 1 - 1
docs/apis/models/detection.md

@@ -71,7 +71,7 @@ train(self, num_epochs, train_dataset, train_batch_size=64, eval_dataset=None, o
 > > - **early_stop** (bool): 是否使用提前终止训练策略。默认为False。
 > > - **early_stop_patience** (int): 当使用提前终止训练策略时,如果验证集精度在`early_stop_patience`个epoch内连续下降或持平,则终止训练。默认为5。
 > > - **use_vdl** (bool): 是否使用VisualDL进行可视化。默认为True。
-> > - **resume_checkpoint** (str): 恢复训练时指定上次训练保存的模型路径,例如`output/ppyolov2/best_model`。若为None,则不会恢复训练。默认值为None。
+> > - **resume_checkpoint** (str): 恢复训练时指定上次训练保存的模型路径,例如`output/ppyolov2/best_model`。若为None,则不会恢复训练。默认值为None。**恢复训练需要将`pretrain_weights`设置为None。**
 
 ### <h3 id="12">evaluate</h3>
 

+ 1 - 1
docs/apis/models/instance_segmentation.md

@@ -63,7 +63,7 @@ train(self, num_epochs, train_dataset, train_batch_size=64, eval_dataset=None, o
 > > - **early_stop** (bool): 是否使用提前终止训练策略。默认为False。
 > > - **early_stop_patience** (int): 当使用提前终止训练策略时,如果验证集精度在`early_stop_patience`个epoch内连续下降或持平,则终止训练。默认为5。
 > > - **use_vdl** (bool): 是否使用VisualDL进行可视化。默认为True。
-> > - **resume_checkpoint** (str): 恢复训练时指定上次训练保存的模型路径,例如`output/maskrcnn_r50fpn/best_model`。若为None,则不会恢复训练。默认值为None。
+> > - **resume_checkpoint** (str): 恢复训练时指定上次训练保存的模型路径,例如`output/maskrcnn_r50fpn/best_model`。若为None,则不会恢复训练。默认值为None。**恢复训练需要将`pretrain_weights`设置为None。**
 
 ### <h3 id="12">evaluate</h3>
 

+ 1 - 1
docs/apis/models/semantic_segmentation.md

@@ -56,7 +56,7 @@ train(num_epochs, train_dataset, train_batch_size=2, eval_dataset=None, optimize
 > > - **early_stop** (bool): 是否使用提前终止训练策略。默认为False。
 > > - **early_stop_patience** (int): 当使用提前终止训练策略时,如果验证集精度在`early_stop_patience`个epoch内连续下降或持平,则终止训练。默认为5。
 > > - **use_vdl** (bool): 是否使用VisualDL进行可视化。默认为True。
-> > - **resume_checkpoint** (str): 恢复训练时指定上次训练保存的模型路径,例如`output/deeplabv3p/best_model`。若为None,则不会恢复训练。默认值为None。
+> > - **resume_checkpoint** (str): 恢复训练时指定上次训练保存的模型路径,例如`output/deeplabv3p/best_model`。若为None,则不会恢复训练。默认值为None。**恢复训练需要将`pretrain_weights`设置为None。**
 
 ### <h3 id="12">evaluate</h3>