Fix default # diffusion samples

wjj2024 · May 20, 2024 · 4c731c5 · 4c731c5
1 parent a1ddcf4
commit 4c731c5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/octo/model/components/action_heads.py b/octo/model/components/action_heads.py
@@ -409,7 +409,7 @@ class DiffusionActionHead(nn.Module):
     hidden_dim: int = 256
     use_layer_norm: bool = True
     diffusion_steps: int = 20
-    n_diffusion_samples: int = 4
+    n_diffusion_samples: int = 1
 
     def setup(self):
         if self.use_map: