Aufgrund einer Störung des s3 Storage, könnten in nächster Zeit folgende GitLab Funktionen nicht zur Verfügung stehen: LFS, Container Registry, Job Artifacs, Uploads (Wiki, Bilder, Projekt-Exporte). Wir bitten um Verständnis. Es wird mit Hochdruck an der Behebung des Problems gearbeitet. Weitere Informationen zur Störung des Object Storage finden Sie hier:

Commit ed52ffa0 authored by Sebastian N.'s avatar Sebastian N.
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">
${} = ${input}
${} = F.ndarray.argmax(${input}, keepdims=True)
\ 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