train_group.md 1.5 KB
Newer Older
W
wuzewu 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13
# cfg.TRAIN

TRAIN Group存放所有和训练相关的配置

## `MODEL_SAVE_DIR`
在训练周期内定期保存模型的主目录

## 默认值
无(需要用户自己填写)

<br/>
<br/>

W
wuzewu 已提交
14
## `PRETRAINED_MODEL_DIR`
W
wuzewu 已提交
15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30
预训练模型路径

## 默认值


## 注意事项

* 若未指定该字段,则模型会随机初始化所有的参数,从头开始训练

* 若指定了该字段,但是路径不存在,则参数加载失败,仍然会被随机初始化

* 若指定了该字段,且路径存在,但是部分参数不存在或者shape无法对应,则该部分参数随机初始化

<br/>
<br/>

W
wuzewu 已提交
31 32
## `RESUME_MODEL_DIR`
从指定路径中恢复参数并继续训练
W
wuzewu 已提交
33 34

## 默认值
W
wuzewu 已提交
35

W
wuzewu 已提交
36 37 38

## 注意事项

W
wuzewu 已提交
39
*`RESUME_MODEL_DIR`存在时,PaddleSeg会恢复到上一次训练的最近一个epoch,并且恢复训练过程中的临时变量(如已经衰减过的学习率,Optimizer的动量数据等),`PRETRAINED_MODEL`路径的最后一个目录必须为int数值或者字符串final,PaddleSeg会将int数值作为当前起始EPOCH继续训练,若目录为final,则不会继续训练。若目录不满足上述条件,PaddleSeg会抛出错误。
W
wuzewu 已提交
40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55

<br/>
<br/>

## `SYNC_BATCH_NORM`
是否在多卡间同步BN的均值和方差

## 默认值
False

## 注意事项

* 打开该选项会带来一定的性能消耗(多卡间同步数据导致)

* 仅在GPU多卡训练时该开关有效(Windows不支持多卡训练,因此无需打开该开关)

W
wuzewu 已提交
56
* GPU多卡训练时,建议开启该开关,可以提升模型的训练效果