一种针对训练任务扩容资源不兼容的回滚方法及装置

摘要

本发明提供了一种针对训练任务扩容资源不兼容的回滚方法及装置，所述的方法包括：提交训练任务脚本时，设置一个位置标签；建立pod，执行训练任务；当进行资源扩容后，判断扩容资源是否兼容，若否，进入下一环节；删除扩容资源，新建一个pod；从位置标签处再次加载并执行训练任务。所述的装置包括：第一设置模块，用于在提交训练任务脚本时设置一个位置标签；第一判断模块，用于判断扩容资源是否兼容，若否，转入回滚模块；回滚模块，用于从位置标签处再次加载训练任务。本申请在发生扩容资源不兼容时，能够自动完成训练任务的自动回滚，大大提高了训练效率。