微比恩 > 信息聚合 > 迷你版 DALL-E:模型缩小 27 倍,训练成本仅 200 美元,在线可玩

迷你版 DALL-E:模型缩小 27 倍,训练成本仅 200 美元,在线可玩

2021-08-02 10:52:41来源: IT之家

自从今年 1 月份 OpenAI 祭出 120 亿参数魔法模型 DALL・E,众多科研工作者开始对其进行多方位的剖析,希望也能做出给定“自然语言文字描述直接生成对应图片”的程序或软件。120 亿参数......... 这意味着只有“顶级玩家”才能入局。如果缩小模型参数,程序还能跑出高质量的结果么?今日,谷歌和 Hugging Face 的研究员们开放了一款 DALL・E mini 的小程序。顾名思义,作为 DALL・E 的复刻版,模型规模比原版小了 27 倍,且只在单个 TPU v3-8 上训练了 3 天。mini 版的 DALL・E 创造思路非常简单:在官方开源代码以及预训练模型的基础上,简化体系结构和减少模型内存需求。效果如何?作者在博客中说:“在硬件资源受限的情况下,虽然质量比较低,但结果仍然是令人印象深刻的。”AI 科技评论将其与正版 DALL・E 进行了对比,结果如下:文本提示:穿着芭蕾舞短裙的萝卜宝宝在遛狗。AI 生

关注公众号