提交 ec85081f 编写于 作者: R rensilin

fix

Change-Id: I69ec0b9d21b37aa5a9671c343b8927948bc14735
上级 8451e1e6
...@@ -19,6 +19,27 @@ class ModelBuilder: ...@@ -19,6 +19,27 @@ class ModelBuilder:
""" """
Attributes: Attributes:
_save_path: Save path of programs _save_path: Save path of programs
def _inference():
Build inference network(without loss and optimizer)
**This function is declared in the network_desc_path file, and will be set in initialize()**
Returns:
list<Variable>: inputs
and
list<Variable>: outputs
pass
def _loss_function(outputs):
**This function is declared in the network_desc_path file, and will be set in initialize()**
Args:
outputs: the second result of inference()
Returns:
Variable: loss
and
list<Variable>: labels
pass
""" """
def initialize(self, network_desc_path, save_path=None): def initialize(self, network_desc_path, save_path=None):
...@@ -68,31 +89,6 @@ class ModelBuilder: ...@@ -68,31 +89,6 @@ class ModelBuilder:
return True return True
def _inference():
"""Build inference network(without loss and optimizer)
**This function is declared in the network_desc_path file, and will be set in initialize()**
Returns:
list<Variable>: inputs
and
list<Variable>: outputs
"""
pass
def _loss_function(outputs):
"""
**This function is declared in the network_desc_path file, and will be set in initialize()**
Args:
outputs: the second result of inference()
Returns:
Variable: loss
and
list<Variable>: labels
"""
pass
def build_and_save(self): def build_and_save(self):
"""Build programs and save to _save_path """Build programs and save to _save_path
""" """
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册