Skip to content
Snippets Groups Projects
Commit ed52ffa0 authored by Sebastian Nickels's avatar Sebastian Nickels
Browse files

Removed default time parameter from models, added ArgMax implementation

parent edb10eba
No related branches found
No related tags found
1 merge request!23Added Unroll-related features and layers
Pipeline #180214 failed
<#assign input = element.inputs[0]>
<#if mode == "FORWARD_FUNCTION">
${element.name} = ${input}
${element.name} = F.ndarray.argmax(${input}, keepdims=True)
</#if>
\ No newline at end of file
......@@ -14,7 +14,7 @@ architecture RNNencdec(max_length=50, vocabulary_size=30000, hidden_size=1000){
encoder.state -> decoder.state;
timed<t=1> GreedySearch(max_length=50) {
timed<t> GreedySearch(max_length=50) {
target[t-1] ->
Embedding(output_dim=hidden_size) ->
decoder ->
......
......@@ -4,7 +4,7 @@ architecture RNNtest(max_length=50, vocabulary_size=30000, hidden_size=1000){
source -> Softmax() -> target[0];
timed <t=1> BeamSearchStart(max_length=5){
timed <t> BeamSearch(max_length=5){
(target[0] | target[t-1]) ->
Concatenate() ->
FullyConnected(units=30000) ->
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment