Exemplo n.º 1
0
Arquivo: t5.py Projeto: 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
Exemplo n.º 2
0
Arquivo: t5.py Projeto: 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
Exemplo n.º 3
0
Arquivo: t5.py Projeto: oceanfly/lit
 def input_spec(self):
     return lit_types.remap_spec(self.model.input_spec(),
                                 self.FIELD_RENAMES)
Exemplo n.º 4
0
Arquivo: t5.py Projeto: PAIR-code/lit
 def input_spec(self):
     return lit_types.remap_spec(self.wrapped.input_spec(),
                                 self.FIELD_RENAMES)