Skip to content
This repository has been archived by the owner on Aug 5, 2022. It is now read-only.

add continuation indicator layer for intel caffe #206

Open
wants to merge 3 commits into
base: master
Choose a base branch
from

Conversation

lddsdu
Copy link

@lddsdu lddsdu commented May 12, 2018

The continuation_indicator_layer is used to produce one blob as the bottom input for lstm layer, and this layer used the parameter below
message ContinuationIndicatorParameter {
optional uint32 time_step = 1 [default = 0];
optional uint32 batch_size = 2 [default = 0];
}
The shape of this blob is (time_step, shape), this may useful when using lstm layer.

@ftian1
Copy link
Contributor

ftian1 commented May 17, 2018

thanks @lddsdu . we usually do same thing by having a python layer. it's mainly because user may not know the time step before data preprocessing. so it's hard to assign a max value in proto file. but anyway your path looks good for me.

@lddsdu
Copy link
Author

lddsdu commented May 18, 2018

@ftian1 Thank you! by the way, I want to know if you would like to add 3d convolution layers in intel caffe, when I using 2d convolution in a 40 cores cpu computer server, it is about 1.5 times as fast as the tesla p4. It's really amazing,I need to use 3d convolution when dealing with videos.

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants