From 307fe75566fe7640a0e2374d6755b85518f183dc Mon Sep 17 00:00:00 2001
From: Edan Toledo <42650996+EdanToledo@users.noreply.github.com>
Date: Sat, 22 Feb 2025 10:58:20 +0000
Subject: [PATCH] chore: revert config

---
 stoix/configs/arch/anakin.yaml | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/stoix/configs/arch/anakin.yaml b/stoix/configs/arch/anakin.yaml
index 594b2330..8d2c40b8 100644
--- a/stoix/configs/arch/anakin.yaml
+++ b/stoix/configs/arch/anakin.yaml
@@ -3,8 +3,8 @@ architecture_name: anakin
 # --- Training ---
 seed: 42  # RNG seed.
 update_batch_size: 1 # Number of vectorised gradient updates per device.
-total_num_envs: 128  # Total Number of vectorised environments across all devices and batched_updates. Needs to be divisible by n_devices*update_batch_size.
-total_timesteps: 1e6 # Set the total environment steps.
+total_num_envs: 1024  # Total Number of vectorised environments across all devices and batched_updates. Needs to be divisible by n_devices*update_batch_size.
+total_timesteps: 1e7 # Set the total environment steps.
 # If unspecified, it's derived from num_updates; otherwise, num_updates adjusts based on this value.
 num_updates: ~ # Number of updates