Couldn't use full power of CUDA cores

YuxianMeng / Matrix-Capsules-pytorch

A Pytorch implementation of "Matrix Capsules with EM routing"

234 stars 48 forks source link

Couldn't use full power of CUDA cores #8

Open FrostbiteXSW opened 4 years ago

FrostbiteXSW commented 4 years ago

When trying your demo on my 1050 Ti, this model only reaches 30% usage of CUDA cores and runs really slow. Meantime, another traditional CapsNet can reach 100% usage. Does this mean a performance problem of your model?

yahiko-l commented 4 years ago

I think many problems not be solved.