hub / github.com/tensorflow/models / create_optimizer

Method create_optimizer

official/modeling/multitask/multitask.py:96–102 · view source on GitHub ↗

(cls,
                       optimizer_config: OptimizationConfig,
                       runtime_config: Optional[RuntimeConfig] = None,
                       dp_config: Optional[DifferentialPrivacyConfig] = None)

Source from the content-addressed store, hash-verified

94
95	@classmethod
96	def create_optimizer(cls,
97	optimizer_config: OptimizationConfig,
98	runtime_config: Optional[RuntimeConfig] = None,
99	dp_config: Optional[DifferentialPrivacyConfig] = None):
100	return base_task.Task.create_optimizer(
101	optimizer_config=optimizer_config, runtime_config=runtime_config,
102	dp_config=dp_config)
103
104	def joint_train_step(self, task_inputs,
105	multi_task_model: base_model.MultiTaskBaseModel,

Callers 2

get_trainerFunction · 0.45

run_experiment_with_multitask_evalFunction · 0.45

Calls

no outgoing calls

Tested by

no test coverage detected