Huawei_Technology
/
mindspore
Not watched
1
0
Fork 0
Code
Releases 13 Wiki evaluate Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain HPC
Browse Source

for pylint 5th

tags/v0.3.0-alpha
z00478463 5 years ago
parent
13321abf7e
commit
78b133b08d
1 changed files with 1 additions and 1 deletions
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. +1
    -1
      example/resnet50_imagenet2012_THOR/model/resnet.py

+ 1
- 1
example/resnet50_imagenet2012_THOR/model/resnet.py View File

@@ -126,7 +126,7 @@ def _bn_last(channel):
def _fc(in_channel, out_channel, damping, loss_scale, frequency): def _fc(in_channel, out_channel, damping, loss_scale, frequency):
weight_shape = (out_channel, in_channel) weight_shape = (out_channel, in_channel)
weight = Tensor(kaiming_uniform(weight_shape, a=math.sqrt(5)) weight = Tensor(kaiming_uniform(weight_shape, a=math.sqrt(5))
return Dense_Thor(in_channel, out_channel, has_bias=False, weight_init=weight, \
return Dense_Thor(in_channel, out_channel, has_bias=False, weight_init=weight,
bias_init=0, damping=damping, loss_scale=loss_scale, frequency=frequency) bias_init=0, damping=damping, loss_scale=loss_scale, frequency=frequency)






Write Preview
Loading…
Cancel
Save