p04896132
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!1237 adopt weight initializator modification in resnet 

Merge pull request !1237 from gengdongjie/master
This commit is contained in:
mindspore-ci-bot 2020-05-19 16:02:54 +08:00 committed by Gitee
parent 45d564d8f7 ae9ce1629b
commit 122a6e03a9
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
example/resnet101_imagenet2012/train.py
View File

@ -64,11 +64,11 @@ if __name__ == '__main__':
if isinstance(cell, nn.Conv2d):
cell.weight.default_input = weight_init.initializer(weight_init.XavierUniform(),
cell.weight.default_input.shape(),
cell.weight.default_input.dtype())
cell.weight.default_input.dtype()).to_tensor()
if isinstance(cell, nn.Dense):
cell.weight.default_input = weight_init.initializer(weight_init.TruncatedNormal(),
cell.weight.default_input.shape(),
cell.weight.default_input.dtype())
cell.weight.default_input.dtype()).to_tensor()
if not config.label_smooth:
config.label_smooth_factor = 0.0
loss = CrossEntropy(smooth_factor=config.label_smooth_factor, num_classes=config.class_num)

4
example/resnet50_imagenet2012/train.py
View File

@ -61,11 +61,11 @@ if __name__ == '__main__':
if isinstance(cell, nn.Conv2d):
cell.weight.default_input = weight_init.initializer(weight_init.XavierUniform(),
cell.weight.default_input.shape(),
cell.weight.default_input.dtype())
cell.weight.default_input.dtype()).to_tensor()
if isinstance(cell, nn.Dense):
cell.weight.default_input = weight_init.initializer(weight_init.TruncatedNormal(),
cell.weight.default_input.shape(),
cell.weight.default_input.dtype())
cell.weight.default_input.dtype()).to_tensor()
if not config.use_label_smooth:
config.label_smooth_factor = 0.0