Skip to content

Commit

Permalink
Update readme.md
Browse files Browse the repository at this point in the history
  • Loading branch information
xwolfs authored Jul 10, 2017
1 parent 1ab1000 commit 2329946
Showing 1 changed file with 19 additions and 4 deletions.
23 changes: 19 additions & 4 deletions readme.md
Original file line number Diff line number Diff line change
Expand Up @@ -12,8 +12,23 @@ Two RNN modules are deployed for temporal dynamic modeling and spatial context m

![Spatial RNN](https://github.com/xwolfs/ResRNN/blob/master/SRNN.pdf)

### ON Circle-RNN
### ON implementation of Circle-RNN

To implement Circle-RNN,the following modifications to the caffe toolbox are required:
1. add 'circle_lstm_layer.hpp' to '/caffe/include/caffe/layers', add 'circle_lstm_layer.cpp' to '/caffe/src/caffe/layers';
2. In 'caffe.proto', add the following line to 'message RecurrentParameter'
'optional uint32 depth = 6 [default = 0];'
Note: LSTM units is employed in Circle-RNN.

#### Example of Circle-RNN
'layer { name: "lstm1" type: "CircleLSTM" bottom: "ip1_permute" bottom: "clip_permute" top: "wt_lstm1"
recurrent_param { num_output: 6 depth: 0
weight_filler { type: "uniform" min: -0.05 max: 0.05 }
bias_filler { type: "constant" value: 0 }
}
}
'
The parameters *depth* indicates how many rounds the RNN network is unrolled.
* *depth=0* : the number of rounds equals the number of time steps of the sequences.
* *depth=1* : the RNN network is unrolled once. This is equivalent to the original RNN in caffe with LSTM unit;

In 'caffe.proto', add the following line to 'message RecurrentParameter'

'optional uint32 depth = 6 [default = 0];'

0 comments on commit 2329946

Please sign in to comment.