diff --git a/crates/ml-core/src/gpu/profile.rs b/crates/ml-core/src/gpu/profile.rs
index c83b17a14..d19b0135d 100644
--- a/crates/ml-core/src/gpu/profile.rs
+++ b/crates/ml-core/src/gpu/profile.rs
@@ -291,9 +291,10 @@ mod tests {
     #[test]
     fn test_embedded_a100_parses() {
         let profile: GpuProfile = toml::from_str(A100_TOML).unwrap();
-        assert_eq!(profile.training.batch_size, 512);
+        assert_eq!(profile.training.batch_size, 2048);
         assert_eq!(profile.training.num_atoms, 51);
-        assert_eq!(profile.training.buffer_size, 200_000);
+        assert_eq!(profile.training.buffer_size, 0);  // 0 = auto from VRAM
+        assert_eq!(profile.experience.gpu_timesteps_per_episode, 500);
         assert_eq!(profile.cuda.cuda_stack_bytes, 65536);
     }
 
@@ -302,7 +303,7 @@ mod tests {
         let profile: GpuProfile = toml::from_str(DEFAULT_TOML).unwrap();
         assert_eq!(profile.training.batch_size, 256);
         assert_eq!(profile.training.num_atoms, 21);
-        assert_eq!(profile.training.buffer_size, 50_000);
+        assert_eq!(profile.training.buffer_size, 0);  // 0 = auto from VRAM
         assert_eq!(profile.experience.gpu_timesteps_per_episode, 200);
         assert_eq!(profile.cuda.cuda_stack_bytes, 32768);
     }