diff --git a/example/config_no_temporal.json b/example/config_no_temporal.json index 962930c..47ef065 100644 --- a/example/config_no_temporal.json +++ b/example/config_no_temporal.json @@ -44,6 +44,6 @@ "cont_clamp_x0": 5.0, "shuffle_buffer": 1024, "use_temporal_stage1": false, - "sample_batch_size": 8, + "sample_batch_size": 4, "sample_seq_len": 128 }