API代码

UAI Train的Keras系统的backend为TensorFlow，因此Keras的API会部分依赖TensorFlow的API

获取方法

git clone https://github.com/ucloud/uai-sdk.git

Keras相关文件路径

uai-sdk/

  examples/
    keras/
       train/
  uaitrain/
    arch/
      tensorflow/
         uflag.py
  uaitrain_tool/
    keras/
      keras_tool.py

简介

uaitrain/arch/tensorflow/uflag.py

该文件为Keras, TensorFlow公用

uaitrain/arch/tensorflow/uflag.py 定义了UAI Train在运行Keras训练任务时所需的参数定义，其中包括固定参数和可变参数。由于Keras使用的是tensorflow的backend，因此可以直接引用TensorFlow的参数

固定参数

参数

默认值

说明

可变参数

参数

默认值

说明

uaitrain_tool/keras/keras_tool.py

keras_tool.py 工具支持镜像打包功能，详细使用方法可参见打包镜像

文档中心