使用前说明
本章节指导用户基于mxRec为用户提供的little-demo样例,快速理解一个使用TF.Session进行模型训练需要准备的相关文件和关键接口适配。little-demo仅是一个代码示例,并介绍了调用相关接口的逻辑,不包含具体的模型,没有实现具体的功能。
little-demo仅作参考学习,不支持在little-demo上适配用户自己的模型。little-demo存放路径为:链接。
在train_and_evaluate场景下不支持多轮eval。
文件名 |
说明 |
---|---|
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 |
精度检测工具。 |
父主题: 快速入门