|
|
|
@@ -126,7 +126,7 @@ def _bn_last(channel): |
|
|
|
def _fc(in_channel, out_channel, damping, loss_scale, frequency): |
|
|
|
weight_shape = (out_channel, in_channel) |
|
|
|
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) |
|
|
|
|
|
|
|
|
|
|
|
|