未验证 提交 bdacb355 编写于 作者: R ruri 提交者: GitHub

add mso significant commit, because of creep API change, no more steps toward...

add mso significant commit, because of creep API change, no more steps toward to be a time killer and all of it is crap (#4953)
上级 7a951e87
...@@ -246,7 +246,7 @@ class TSN_ResNet(paddle.nn.Layer): ...@@ -246,7 +246,7 @@ class TSN_ResNet(paddle.nn.Layer):
self.class_dim, self.class_dim,
weight_attr=ParamAttr( weight_attr=ParamAttr(
initializer=paddle.nn.initializer.Normal( initializer=paddle.nn.initializer.Normal(
loc=0.0, scale=0.01), mean=0.0, std=0.01),
name="fc_0.w_0"), name="fc_0.w_0"),
bias_attr=ParamAttr( bias_attr=ParamAttr(
initializer=paddle.nn.initializer.Constant(value=0.0), initializer=paddle.nn.initializer.Constant(value=0.0),
......
...@@ -161,7 +161,7 @@ def create_optimizer(cfg, params): ...@@ -161,7 +161,7 @@ def create_optimizer(cfg, params):
momentum = cfg.momentum momentum = cfg.momentum
optimizer = paddle.optimizer.Momentum( optimizer = paddle.optimizer.Momentum(
learning_rate=paddle.optimizer.PiecewiseLR( learning_rate=paddle.optimizer.lr.PiecewiseDecay(
boundaries=bd, values=lr), boundaries=bd, values=lr),
momentum=momentum, momentum=momentum,
weight_decay=paddle.regularizer.L2Decay(l2_weight_decay), weight_decay=paddle.regularizer.L2Decay(l2_weight_decay),
...@@ -190,7 +190,7 @@ def train(args): ...@@ -190,7 +190,7 @@ def train(args):
video_model = paddle.DataParallel(video_model) video_model = paddle.DataParallel(video_model)
pre_state_dict, _ = paddle.load(args.pretrain) pre_state_dict = paddle.load(args.pretrain)
#if paddle.distributed.parallel.Env().local_rank == 0: #if paddle.distributed.parallel.Env().local_rank == 0:
video_model = init_model(video_model, pre_state_dict) video_model = init_model(video_model, pre_state_dict)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册