model_group.md 1.6 KB
Newer Older
W
wuzewu 已提交
1 2 3 4 5 6 7
# cfg.MODEL

MODEL Group存放所有和模型相关的配置,该Group还包含三个子Group

* [DeepLabv3p](./model_deeplabv3p_group.md)
* [UNet](./model_unet_group.md)
* [ICNet](./model_icnet_group.md)
W
wuyefeilin 已提交
8
* [HRNet](./model_hrnet_group.md)
W
wuzewu 已提交
9 10 11

## `MODEL_NAME`

W
wuyefeilin 已提交
12
所选模型,支持`deeplabv3p` `unet` `icnet` `hrnet`四种模型
W
wuzewu 已提交
13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114

### 默认值

无(需要用户自己填写)

<br/>
<br/>

## `DEFAULT_NORM_TYPE`

模型所用norm类型,支持`bn` [`gn`]()

### 默认值

`bn`

<br/>
<br/>

## `DEFAULT_GROUP_NUMBER`

默认GROUP数量,仅在`DEFAULT_NORM_TYPE``gn`时生效

### 默认值

32

<br/>
<br/>

## `BN_MOMENTUM`

BatchNorm动量, 一般无需改动

### 默认值

0.99

<br/>
<br/>

## `DEFAULT_EPSILON`

BatchNorm计算时所用的极小值, 防止分母除0溢出,一般无需改动

### 默认值

1e-5

<br/>
<br/>

## `FP16`

是否开启FP16训练

### 默认值

False

<br/>
<br/>

## `SCALE_LOSS`

对损失进行缩放的系数

### 默认值

1.0

### 注意事项
* 启动fp16训练时,建议设置该字段为8

<br/>
<br/>

## `MULTI_LOSS_WEIGHT`

多路损失的权重

### 默认值

[1.0]

### 注意事项

* 该字段仅在模型存在多路损失的情况下生效

* 目前支持的模型中只有`icnet`使用多路(3路)损失

* 当选择模型为`icnet`且该字段的长度不为3时,PaddleSeg会强制设置该字段为[1.0, 0.4, 0.16]

### 示例
假设模型存在三路损失,计算结果分别为loss1/loss2/loss3,并且`MULTI_LOSS_WEIGHT`的值为[1.0, 0.4, 0.16],则最终损失的计算结果为
```math
loss = 1.0 * loss1 + 0.4 * loss2 + 0.16 * loss3
```

<br/>
<br/>