TrainingPodSpecParam

class nemo_microservices.types.TrainingPodSpecParam

Bases: TypedDict

annotations: Dict[str, str]

Additional arguments for annotations

node_affinity: NodeAffinityParam

The kubernentes node affinity to apply to the training pods

node_selectors: Dict[str, str]

Additional arguments for node selector

tolerations: Iterable[TolerationParam]

Additional arguments for tolerations