MCPcopy Create free account
hub / github.com/YunwenTechnology/QueryGeneration / construct_scalar_host_call

Function construct_scalar_host_call

train/utils.py:169–219  ·  view source on GitHub ↗

Construct a host call to log scalars when training on TPU. Args: metric_dict: A dict of the tensors to be logged. model_dir: The location to write the summary. prefix: The prefix (if any) to prepend to the metric names. Returns: A tuple of (function, args_to_be_pass

(metric_dict, model_dir, prefix="")

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected