Skip to content

ai: Add LiME-TM package - #2015

Open
wuhanstudio wants to merge 4 commits into
RT-Thread:masterfrom
wuhanstudio:lime-tm
Open

ai: Add LiME-TM package#2015
wuhanstudio wants to merge 4 commits into
RT-Thread:masterfrom
wuhanstudio:lime-tm

Conversation

@wuhanstudio

Copy link
Copy Markdown

LiME-TM: 在 MCU 上 训练 机器学习模型

https://tm.wuhanstudio.uk/#/8

例程包括训练完整的 MNIST 数据集 (60,000 张图片),训练图片从 SD 卡上一张张读取,模型训练只需要 20KB SRAM。

虽然相比 CNN,模型精度下降了 1%,但是训练速度快了 300x,推理速度快了 20x,并且模型训练比模型推理更快。

因为 Tsetlin Machine 机器学习模型没有浮点运算,不需要梯度;纯逻辑运算,并且利用强化学习 sample-by-sample 训练,所以比 CNN 更加节省 SRAM,并且整形运算也更快。

image

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant