TrainingPodSpec

class nemo_microservices.types.TrainingPodSpec(*args: Any, **kwargs: Any)

Bases: BaseModel

annotations: Dict[str, str] | None = None

Additional arguments for annotations

node_affinity: NodeAffinity | None = None

The kubernentes node affinity to apply to the training pods

node_selectors: Dict[str, str] | None = None

Additional arguments for node selector

tolerations: List[Toleration] | None = None

Additional arguments for tolerations