提交 95cdf430 编写于 作者: Q qjing666

fix bug in submitter_demo

上级 925ace88
......@@ -96,7 +96,7 @@ else:
for data in reader():
trainer.run(feed=data, fetch=[])
step_i += 1
if train_step == trainer._step:
if step_i == trainer._step:
break
epoch_id += 1
if epoch_id % 5 == 0:
......
......@@ -12,6 +12,6 @@
# See the License for the specific language governing permissions and
# limitations under the License.
""" PaddleFL version string """
fl_version = "0.1.9"
module_proto_version = "0.1.9"
fl_version = "0.1.10"
module_proto_version = "0.1.10"
......@@ -29,7 +29,7 @@ def python_version():
max_version, mid_version, min_version = python_version()
REQUIRED_PACKAGES = [
'six >= 1.10.0', 'protobuf >= 3.1.0','paddlepaddle >= 1.6', 'zmq'
'six >= 1.10.0', 'protobuf >= 3.1.0','paddlepaddle >= 1.6', 'zmq', 'paddlepaddle-gpu >= 1.6'
]
if max_version < 3:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册