提交 b5ebb2fc 编写于 作者: Y Yang Zhang

Point out that multi device execution not implemented in dgyraph

上级 f7cf7b5b
...@@ -349,6 +349,7 @@ class DynamicGraphAdapter(object): ...@@ -349,6 +349,7 @@ class DynamicGraphAdapter(object):
def mode(self, value): def mode(self, value):
self.model.mode = value self.model.mode = value
# TODO multi device in dygraph mode not implemented at present time
def train(self, inputs, labels, device='CPU', device_ids=None): def train(self, inputs, labels, device='CPU', device_ids=None):
assert self.model._optimizer and self.model._loss_functions, \ assert self.model._optimizer and self.model._loss_functions, \
"model not ready, please call `model.prepare()` first" "model not ready, please call `model.prepare()` first"
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册