commit
ed600b30d6
|
@ -16,6 +16,7 @@ from __future__ import absolute_import
|
|||
from __future__ import division
|
||||
from __future__ import print_function
|
||||
|
||||
import math
|
||||
import paddle
|
||||
from paddle import nn, ParamAttr
|
||||
from paddle.nn import functional as F
|
||||
|
@ -88,11 +89,14 @@ class LocalizationNetwork(nn.Layer):
|
|||
in_channels = num_filters
|
||||
self.block_list.append(pool)
|
||||
name = "loc_fc1"
|
||||
stdv = 1.0 / math.sqrt(num_filters_list[-1] * 1.0)
|
||||
self.fc1 = nn.Linear(
|
||||
in_channels,
|
||||
fc_dim,
|
||||
weight_attr=ParamAttr(
|
||||
learning_rate=loc_lr, name=name + "_w"),
|
||||
learning_rate=loc_lr,
|
||||
name=name + "_w",
|
||||
initializer=nn.initializer.Uniform(-stdv, stdv)),
|
||||
bias_attr=ParamAttr(name=name + '.b_0'),
|
||||
name=name)
|
||||
|
||||
|
|
Loading…
Reference in New Issue