diff --git a/example/config.json b/example/config.json index 8d7866d..3e2a187 100644 --- a/example/config.json +++ b/example/config.json @@ -7,8 +7,8 @@ "out_dir": "./results", "device": "auto", "timesteps": 600, - "batch_size": 32, - "seq_len": 128, + "batch_size": 16, + "seq_len": 96, "epochs": 10, "max_batches": 4000, "lambda": 0.7,