fix: fix comments

pull/1602/head
gaotingquan 2021-12-20 06:55:40 +00:00 committed by Tingquan Gao
parent f8ee6c0f86
commit 41296972a4
1 changed files with 6 additions and 6 deletions

View File

@ -119,8 +119,8 @@ class TheseusLayer(nn.Layer):
"""use 'replace_function' to modify the 'layer_name_pattern'.
Args:
layer_name_pattern (str): The name of target layer variable.
replace_function (FunctionType): The function to modify target layer,
layer_name_pattern (Union[str, List[str]]): The name of layer to be modified by 'replace_function'.
replace_function (FunctionType): The function to modify target layer specified by 'layer_name_pattern'.
Returns:
bool: 'True' if successful, 'False' otherwise.
@ -156,7 +156,7 @@ class TheseusLayer(nn.Layer):
"""stop forward and backward after 'stop_layer_name'.
Args:
stop_layer_name (str): The name of target layer variable.
stop_layer_name (str): The name of layer that stop forward and backward after this layer.
Returns:
bool: 'True' if successful, 'False' otherwise.
@ -193,13 +193,13 @@ class TheseusLayer(nn.Layer):
def update_res(self,
return_patterns: Union[str, List[str]]) -> Dict[str, bool]:
"""update the results needed returned.
"""update the results to be returned.
Args:
return_patterns (Union[str, List[str]]): [description]
return_patterns (Union[str, List[str]]): The name of layer to return output.
Returns:
Dict[str, bool]: The pattern(str) is be set successfully if True(bool), failed otherwise.
Dict[str, bool]: The pattern(str) is be set successfully if 'True'(bool), failed if 'False'(bool).
"""
class Handler(object):