Python remap_spec示例

示例#1

文件： t5.py 项目： oceanfly/lit

 def output_spec(self):
     spec = lit_types.remap_spec(self.model.output_spec(),
                                 self.FIELD_RENAMES)
     spec["rougeL"] = lit_types.Scalar()
     return spec

示例#2

文件： t5.py 项目： oceanfly/lit

 def input_spec(self):
     spec = lit_types.remap_spec(self.model.input_spec(),
                                 self.FIELD_RENAMES)
     spec["source_language"] = lit_types.CategoryLabel()
     spec["target_language"] = lit_types.CategoryLabel()
     return spec

示例#3

文件： t5.py 项目： oceanfly/lit

 def input_spec(self):
     return lit_types.remap_spec(self.model.input_spec(),
                                 self.FIELD_RENAMES)

示例#4

文件： t5.py 项目： PAIR-code/lit

 def input_spec(self):
     return lit_types.remap_spec(self.wrapped.input_spec(),
                                 self.FIELD_RENAMES)