文档
注册

使用前说明

本章节指导用户基于mxRec为用户提供的little-demo样例,快速理解一个使用TF.Session进行模型训练需要准备的相关文件和关键接口适配。little-demo仅是一个代码示例,并介绍了调用相关接口的逻辑,不包含具体的模型,没有实现具体的功能。

little-demo仅作参考学习,不支持在little-demo上适配用户自己的模型。little-demo存放路径为:链接

在train_and_evaluate场景下不支持多轮eval。

表1 little-demo文件说明

文件名

说明

config.py

模型相关配置。

dataset.py

数据集预处理。

main.py

模型训练入口。

model.py

模型搭建。

op_impl_mode.ini

算子配置文件。

optimizer.py

优化器。

random_data_generator.py

数据集随机生成。

run.sh

模型训练启动脚本。

run_model.py

训练、推理流程封装。

hccl_json_1p.json

使用通过配置文件设置资源信息的方式启动训练任务(单卡配置样例)。

hccl_json_4p.json

使用通过配置文件设置资源信息的方式启动训练任务(四卡配置样例)。

hccl_json_8p.json

使用通过配置文件设置资源信息的方式启动训练任务(八卡配置样例)。

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词