Module: tfm.core.base_task

Defines the base task abstraction.

Classes

class DifferentialPrivacyConfig

class OptimizationConfig: Configuration for optimizer and learning rate schedule.

class RuntimeConfig: High-level configurations for Runtime.

class Task: A single-replica view of training procedure.