创建模型
训练离线模型,利用本地算力,平台仅记录训练过程中的参数、指标及实验过程。
在平台上实现 Offline-Training,进行创建模型的方式分为以下两种:
在线创建模型,直接获取到模型ID。
线下使用SDK创建模型,获取到模型ID。
在线创建模型
点击顶部导航中的【项目】,进入当前账号的项目界面。
查看当前所有的项目列表,选择目标项目,点击进入该项目界面。
进入项目详情页,查看该项目相关信息。
点击左侧导航栏,点击【模型】,进入该项目的模型列表页。
点击右上角【添加模型】,填写相关信息。
- 模型名称:自定义模型名称。
- Git地址:离线训练不需要填写git项目地址。
- 模型权限:选择 “私有” 或 “公开”。默认为 “私有”。备注:(1)如何编写本地训练代码格式,详情请参考【开发标准格式的训练模型】;(2)将本地项目文件提交git或其他存储库以及相关配置问题, 详情请了解【git项目开发入门】和【项目关联配置】 。
点击确定后,进入模型概览页。
使用SDK创建模型
登录平台账号
import waveletai
waveletai.init(api_token="YOUR_TOKEN")备注:(1)如何使用api_token进行平台登录,详情请参考【waveletai-平台账号登录】。
登录平台,点击目标项目,进入项目列表页,查看模型App的id。
创建模型
函数
waveletai.create_model()
waveletai.create_model(name, desc=None, git_url=None, http_url=None, visibility=Visibility.PRIVATE.value)
参数说明:
参数 类型 详情说明 name str 模型名称 desc str 模型备注 git_url str git地址 http_url str http地址 visibility public/private,默认private 示例:
# 创建模型
app_id = '0b25303a6bee486caff1a03158d43040'
# 设置当前所在的项目
waveletai.set_app(app_id)
# 创建模型
model_name = waveletai.create_model("model1", desc="示例模型", git_url=None, http_url=None)
print("model_name: ", model_name.id)登录平台,查看创建好的模型。
至此,模型创建成功。