Brutex/cavis
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

More test fixes

Browse Source 
Signed-off-by: brian <brian@brutex.de>
This commit is contained in:
Brian Rosenberger 2022-10-07 12:50:54 +02:00
parent b8a21bc991
commit a7f75fe6db
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
cavis-zoo/cavis-zoo-models/src/main/java/org/deeplearning4j/zoo/model/TextGenerationLSTM.java
View File

@ -47,7 +47,7 @@ public class TextGenerationLSTM extends ZooModel {
@Builder.Default private long seed = 1234;
@Builder.Default private int maxLength = 40;
@Builder.Default private int totalUniqueCharacters = 47;
@Builder.Default private int[] inputShape = new int[] {maxLength, totalUniqueCharacters};
@Builder.Default private int[] inputShape = new int[] {40, 47};
@Builder.Default private IUpdater updater = new RmsProp(0.01);
@Builder.Default private CacheMode cacheMode = CacheMode.NONE;
@Builder.Default private WorkspaceMode workspaceMode = WorkspaceMode.ENABLED;