MCPcopy Create free account
hub / github.com/tensorflow/lingvo / WriteInferenceGraph

Method WriteInferenceGraph

lingvo/trainer.py:764–866  ·  view source on GitHub ↗

Generates the inference graphs for a given model. Args: cfg: Full `~.hyperparams.Params` for the model class. If present, this cfg will be used instead of retrieving from model_registry. prune_graph: If true, prune the graph to just the parts we need. Returns: Inf

(self, cfg=None, prune_graph=True)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 4

StartMethod · 0.95

Calls 3

GetParamsMethod · 0.80
TaskNamesMethod · 0.80
ExportMethod · 0.45