文件名称:
slot-attention:GoogleAI实现广告位关注-源码
开发工具:
文件大小: 121kb
下载次数: 0
上传时间: 2021-03-18
详细说明:插槽注意
实施从纸“对象为中心的学习与插槽警示”中Pytorch。这是一段介绍此网络功能的。
更新:官方存储库已在发布
安装
$ pip install slot_attention
用法
import torch
from slot_attention import SlotAttention
slot_attn = SlotAttention (
num_slots = 5 ,
dim = 512 ,
iters = 3 # iterations of attention, defaults to 3
)
inputs = torch . randn ( 2 , 1024 , 512 )
slot_attn ( inputs ) # (2, 5, 512)
训练后,据报道该网络能够推广到稍有不同数量的插槽(群集)。您可以向前覆盖num_slots关键字使用的
(系统自动生成,下载前可以参看下载内容)
下载文件列表
相关说明
- 本站资源为会员上传分享交流与学习,如有侵犯您的权益,请联系我们删除.
- 本站是交换下载平台,提供交流渠道,下载内容来自于网络,除下载问题外,其它问题请自行百度。
- 本站已设置防盗链,请勿用迅雷、QQ旋风等多线程下载软件下载资源,下载后用WinRAR最新版进行解压.
- 如果您发现内容无法下载,请稍后再次尝试;或者到消费记录里找到下载记录反馈给我们.
- 下载后发现下载的内容跟说明不相乎,请到消费记录里找到下载记录反馈给我们,经确认后退回积分.
- 如下载前有疑问,可以通过点击"提供者"的名字,查看对方的联系方式,联系对方咨询.