提交 cd330578 编写于 作者: Y yi.wu

add num_passes

上级 8370c5af
...@@ -75,7 +75,8 @@ def get_model(args): ...@@ -75,7 +75,8 @@ def get_model(args):
shapes=[[-1, 1, 28, 28], (-1, 1)], shapes=[[-1, 1, 28, 28], (-1, 1)],
lod_levels=[0, 0], lod_levels=[0, 0],
dtypes=["float32", "int64"], dtypes=["float32", "int64"],
thread_num=args.gpus) thread_num=args.gpus,
pass_num=args.pass_num)
data_file = fluid.layers.double_buffer( data_file = fluid.layers.double_buffer(
fluid.layers.batch( fluid.layers.batch(
data_file, batch_size=args.batch_size)) data_file, batch_size=args.batch_size))
......
...@@ -157,7 +157,8 @@ def get_model(args): ...@@ -157,7 +157,8 @@ def get_model(args):
shapes=[[-1] + dshape, (-1, 1)], shapes=[[-1] + dshape, (-1, 1)],
lod_levels=[0, 0], lod_levels=[0, 0],
dtypes=["float32", "int64"], dtypes=["float32", "int64"],
thread_num=args.gpus) thread_num=args.gpus,
pass_num=args.pass_num)
data_file = fluid.layers.double_buffer( data_file = fluid.layers.double_buffer(
fluid.layers.batch( fluid.layers.batch(
data_file, batch_size=args.batch_size)) data_file, batch_size=args.batch_size))
......
...@@ -75,7 +75,8 @@ def get_model(args): ...@@ -75,7 +75,8 @@ def get_model(args):
shapes=[[-1] + data_shape, (-1, 1)], shapes=[[-1] + data_shape, (-1, 1)],
lod_levels=[0, 0], lod_levels=[0, 0],
dtypes=["float32", "int64"], dtypes=["float32", "int64"],
thread_num=args.gpus) thread_num=args.gpus,
pass_num=args.pass_num)
data_file = fluid.layers.double_buffer( data_file = fluid.layers.double_buffer(
fluid.layers.batch( fluid.layers.batch(
data_file, batch_size=args.batch_size)) data_file, batch_size=args.batch_size))
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册