昇腾社区首页
中文
注册

使用前说明

本章节提供一个little-demo用例指导用户基于Rec SDK Torch搭建模型。little-demo存放路径为:链接

表1 little-demo文件说明

文件名

说明

main.py

模型训练入口

dataset.py

数据集生成

model.py

模型文件

bash.sh

启动脚本

README.md

demo模型运行说明