search

szagoruyko / imagine-nn

IMAGINE torch neural network routines
Other
109 stars 35 forks source link

thoughts on porting averagepooling ceil mode into opencl? #15

Closed hughperkins closed 9 years ago

hughperkins commented 9 years ago

Hi Sergey,

Looks like some neural-style models could plausibly benefit from averagepooling. Therefore needs ceil mode. So, I'm pondering vaguely porting your inn averagepooling to cl somehow/somewhere. Seems like the easiest way to do this might be to fork inn itself, eg to 'clinn'? (Otherwise needs to port everything into nn, which whilst a good long-term goal, might take more than a few days probably I would think?)

Hugh

(PS I suppose one other way would be to port directly into clnn, and then hack neural-style's loadcaffe_wrapper.lua, that might be easier actually.... still pondering...)

szagoruyko commented 9 years ago

Hi Hugh, check this https://github.com/torch/cunn/pull/134 it will go into nn&cunn soon

hughperkins commented 9 years ago

Awesomeness! Thanks! :-)