naturomics / CapsNet-Tensorflow

A Tensorflow implementation of CapsNet(Capsules Net) in paper Dynamic Routing Between Capsules
Apache License 2.0
3.8k stars 1.17k forks source link

distributed version and acceralation #24

Closed www0wwwjs1 closed 6 years ago

www0wwwjs1 commented 6 years ago

群主你好,我就用中文节省点时间了。

这是我写的一个分布式版本,基本上写了一大半了。为了分布式工作,我把参数都封装到slim里面去了,然后对caps层做了一些优化,我看了你最新的版本,你应该是把xifeng版本primary cap层的结构吸收过来了,实际上digits cap层的结构也可以优化。

我在一个gpu上跑了一下我这个结构的前向,100次大概耗时8s,你的版本大概是23s。

xifeng的版本我也跑过,他使用keras封装的,据他说速度可以快不少,但是我跑下来实际上跟你的版本差不多,总之不是很稳定。所以我用纯tf重写了一遍,供参考,这两天我会把分布式版本补完,到时候进行完整的速度和性能测试。

速度是我认为现在比较关键的一个问题,以后常讨论啊!