下载
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

使用前说明

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

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

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

表1 little-demo文件说明

文件名

说明

config.py

模型相关配置。

dataset.py

数据集预处理。

deterministic_loss

确定性计算loss样例

main.py

模型训练入口。

model.py

模型搭建。

op_impl_mode.ini

算子配置文件。

optimizer.py

优化器。

random_data_generator.py

数据集随机生成。

README.md

demo模型运行说明。

run_deterministic.sh

运行确定性计算的脚本。

run_model.py

训练、推理流程封装。

run.sh

模型训练启动脚本。

utils.py

精度检测工具。

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

当前产品无相关内容

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