mmpretrain增加回归任务组件


mmpretrain 增加回归任务组件

mmpretrain 原生只服务分类任务。本文把它魔改出完整的回归能力:自定义数据集、预处理器、回归器、回归头、MAE 损失与评估指标、推理器,一路改到 gradio 演示,全程沿 mm 系列的注册器机制进行,不破坏框架原有结构。

openmmlab介绍

openmmlab系列框架在2023年全面进入2.0时代,官方把自监督mmselfsup和分类mmclassification两个项目合并成mmpretrain。
openmmlab 2.0系列底层是有一些变化的,最大变化是从mmcv中抽离出功能更强大的mmengine。新的项目我还没仔细看过,目...

Read more

nvidia TAO Toolkit使用总结


nvidia TAO Toolkit 使用总结

TAO 表示 train、adapt、optimize。TAO Toolkit 是 nvidia TAO 框架的低代码版本,主打使用者无需专业 AI 知识即可构建高准确度模型,在英伟达设备上做到训练部署一条龙。低代码不等于无坑,本文按流程走一遍,把每一步的暗雷标出来。

流程参考快速上手 tao toolkit

环境准备

1. 注册与登录 ngc

注册英伟达开发者账号(略),去 ngc 官网目录右上角用英伟达账号登录。

2. 生成 API Key

回到 ngc 首页,右上角点自己的账号,进 Setup,出现 Generate API Ke...

Read more