Commit ed52ffa0 authored by Sebastian Nickels's avatar Sebastian Nickels
Browse files

Removed default time parameter from models, added ArgMax implementation

parent edb10eba
Pipeline #180214 failed with stages
in 2 minutes and 26 seconds
<#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) ->
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment