o #j@sdZddlmZGdddeZGdddeZGdddeZGd d d eZGd d d eZGd ddeZ GdddeZ GdddeZ GdddeZ dS)z AIHC job model module. )Optionalc*eZdZdZdedeffdd ZZS)Labelu 训练任务标签 keyvaluec"tt|||d<||d<dS)ut 初始化训练任务标签 Args: key: 标签键 value: 标签值 rrN)superr__init__)selfrr __class__m/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/baidubce/services/aihc/modules/job/job_model.pyr  zLabel.__init____name__ __module__ __qualname____doc__strr __classcell__r r r rrsrcs&eZdZdZdeffdd ZZS)Optionu% 数据卷挂载选项信息 readOnlycstt|||d<dS)u} 初始化数据卷挂载选项 Args: readOnly: 是否以只读模式挂载到容器中 rN)rrr )r rr r rr *s zOption.__init__rr r r rr%src sDeZdZdZ  d dedededeedeef fdd ZZS) DatasourceuQ 训练任务数据源配置 支持类型：pfs/hostPath/dataset/bos Ntypename mountPath sourcePathoptionscsNtt|||d<||d<||d<|dur||d<|dur%||d<dSdS)u 初始化数据源配置 Args: type: 数据源类型，枚举值：pfs/hostPath/dataset/bos name: 数据源名称（pfs类型时为实例id，bos类型可为空） mountPath: 容器内挂载路径 sourcePath: 源路径（pfs时为存储路径，hostpath时为宿主机路径）(可选) options: 数据源参数（可选） rrrNrr)rrr )r rrrrrr r rr ;s zDatasource.__init__NN) rrrrrrdictr rr r r rr5s rcs4eZdZdZddeedeeffdd ZZS)TensorboardConfigu! tensorboard相关配置 NenablelogPqthc6tt||dur||d<|dur||d<dSdS)u 初始化tensorboard配置 Args: enable: 是否开启Tensorboard（可选） logPqth: 训练任务Tensorboard日志（可选） Nr"ZlogPath)rr!r )r r"r#r r rr [  zTensorboardConfig.__init__r) rrrrrboolrr rr r r rr!Ws(r!csteZdZdZ        d deedeedeedeedeedeed eed eeffd d ZZS) AlertConfigu 告警相关配置 NalertIds instanceId alertName alertItemsfor_ description notifyRuleIdseverityc st|dur ||d<|dur||d<|dur||d<|dur%||d<|dur-||d<|dur5||d<|dur=||d<|durG||d <dSdS) u' 初始化告警配置 Nr(r)r*r+forr-r.r/)rr ) r r(r)r*r+r,r-r.r/r r rr os$  zAlertConfig.__init__)NNNNNNNN)rrrrrrr rr r r rr'js6 r'c sVeZdZdZ     d deedeedeedeedeef fdd ZZ S) JobSpecu" 创建训练任务配置 N imageConfig resourcesenvs enableRDMA hostNetworkcsvtt|||d<||d<|dur||d<|dur||d<|dur'||d<|dur/||d<|dur9||d<dSdS) u 初始化训练任务配置 Args: image: 镜像名称 replicas: 副本数量 imageConfig: 镜像配置（可选） resources: 资源配置（可选） envs: 环境变量（可选） enableRDMA: 是否启用RDMA（可选） hostNetwork: 是否使用主机网络（可选） imagereplicasNr2r3r4r5r6)rr1r )r r7r8r2r3r4r5r6r r rr s zJobSpec.__init__)NNNNN) rrrrrr listr&r rr r r rr1s$r1cs eZdZdZfddZZS) ImageConfigu 任务镜像配置，仅私有镜像时需要配置。 被创建训练任务接口引用。 :param username: 私有镜像仓库用户名 :param password: 私有镜像仓库密码 cr)u 初始化镜像配置 Args: username: 私有镜像仓库用户名 password: 私有镜像仓库密码 usernamepasswordN)rr:r )r r;r<r r rr rzImageConfig.__init__rrrrr rr r r rr:sr:cr)ResourceuZ 任务资源配置 被如下接口引用：创建训练任务 资源名称示例（具体以平台最新文档为准）： - baidu.com/a800_80g_cgpu：GPU型号（需按百度资源描述符填写） - cpu：CPU配额，单位核 - memory：内存配额，单位GB - sharedMemory：共享内存配额，单位GB rquantitycr)u 初始化资源配置 Args: name: 资源名称，示例值： - baidu.com/a800_80g_cgpu：GPU型号 - cpu：CPU配额 - memory：内存配额 - sharedMemory：共享内存配额 quantity: 资源量，字符串类型 示例："8"（表示8核CPU/8GB内存等） rr?N)rr>r )r rr?r r rr s  zResource.__init__rr r r rr>s r>cs"eZdZdZdfdd ZZS)Envu 环境变量信息，被创建训练任务、查询训练任务详情接口引用。 Attributes: name: 标签名（可选） value: 标签值（可选） Ncr$)u 初始化环境变量 Args: name: 环境变量名（可选） value: 环境变量值（可选） Nrr)rr@r )r rrr r rr r%z Env.__init__rr=r r r rr@sr@N) rtypingrr rrrr!r'r1r:r>r@r r r rs "&)