update lsc param name

This commit is contained in:
GuoxiaWang
2021-10-18 12:02:58 +08:00
parent 0c6ba50293
commit 2896bb9fc9
2 changed files with 2 additions and 2 deletions

View File

@@ -58,7 +58,7 @@ class LargeScaleClassifier(nn.Layer):
self._parameter_list = []
if name is None:
name = 'dist@fc@rank@%05d.w' % rank
name = 'dist@fc@rank@%05d' % rank
stddev = math.sqrt(2.0 / (self.embedding_size + self.num_local))
param_attr = paddle.ParamAttr(

View File

@@ -64,7 +64,7 @@ class LargeScaleClassifier(object):
self.output_dict = OrderedDict()
if name is None:
name = 'dist@fc@rank@%05d.w' % rank
name = 'dist@fc@rank@%05d' % rank
stddev = math.sqrt(2.0 / (self.embedding_size + self.num_local))
param_attr = paddle.ParamAttr(