More test fixes

Signed-off-by: brian <brian@brutex.de>
master
Brian Rosenberger 2022-10-07 12:50:54 +02:00
parent b8a21bc991
commit a7f75fe6db
1 changed files with 1 additions and 1 deletions

View File

@ -47,7 +47,7 @@ public class TextGenerationLSTM extends ZooModel {
@Builder.Default private long seed = 1234; @Builder.Default private long seed = 1234;
@Builder.Default private int maxLength = 40; @Builder.Default private int maxLength = 40;
@Builder.Default private int totalUniqueCharacters = 47; @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 IUpdater updater = new RmsProp(0.01);
@Builder.Default private CacheMode cacheMode = CacheMode.NONE; @Builder.Default private CacheMode cacheMode = CacheMode.NONE;
@Builder.Default private WorkspaceMode workspaceMode = WorkspaceMode.ENABLED; @Builder.Default private WorkspaceMode workspaceMode = WorkspaceMode.ENABLED;