提交 cfdf5e23 编写于 作者: J jiangjiajun

Merge branch 'develop' of https://github.com/PaddlePaddle/PaddleX into develop

...@@ -71,7 +71,7 @@ class HRNet(object): ...@@ -71,7 +71,7 @@ class HRNet(object):
self.end_points = [] self.end_points = []
return return
def net(self, input, class_dim=1000): def net(self, input):
width = self.width width = self.width
channels_2, channels_3, channels_4 = self.channels[width] channels_2, channels_3, channels_4 = self.channels[width]
num_modules_2, num_modules_3, num_modules_4 = 1, 4, 3 num_modules_2, num_modules_3, num_modules_4 = 1, 4, 3
...@@ -125,7 +125,7 @@ class HRNet(object): ...@@ -125,7 +125,7 @@ class HRNet(object):
stdv = 1.0 / math.sqrt(pool.shape[1] * 1.0) stdv = 1.0 / math.sqrt(pool.shape[1] * 1.0)
out = fluid.layers.fc( out = fluid.layers.fc(
input=pool, input=pool,
size=class_dim, size=self.num_classes,
param_attr=ParamAttr( param_attr=ParamAttr(
name='fc_weights', name='fc_weights',
initializer=fluid.initializer.Uniform(-stdv, stdv)), initializer=fluid.initializer.Uniform(-stdv, stdv)),
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册