This commit is contained in:
2025-11-21 02:57:50 -05:00
parent 8a7f9570c1
commit e6b5e43646
2 changed files with 15 additions and 1 deletions

View File

@@ -340,7 +340,7 @@ class Train(Train_Base):
def train(self, args): def train(self, args):
# --------------------------------------- Learning parameters # --------------------------------------- Learning parameters
n_envs = min(12, os.cpu_count()) n_envs = min(8, os.cpu_count())
n_steps_per_env = 1024 # RolloutBuffer is of size (n_steps_per_env * n_envs) n_steps_per_env = 1024 # RolloutBuffer is of size (n_steps_per_env * n_envs)
minibatch_size = 64 # should be a factor of (n_steps_per_env * n_envs) minibatch_size = 64 # should be a factor of (n_steps_per_env * n_envs)
total_steps = 50000000 total_steps = 50000000

14
train.sh Executable file
View File

@@ -0,0 +1,14 @@
#!/bin/bash
source /home/lumi/Downloads/Anaconda/etc/profile.d/conda.sh
conda activate FCP
export MKL_SERVICE_FORCE_INTEL=1
export MKL_THREADING_LAYER=GNU
export LD_LIBRARY_PATH="/usr/lib/x86_64-linux-gnu:$LD_LIBRARY_PATH"
pkill -f rcssserver3d 2>/dev/null || true
pkill -f simspark 2>/dev/null || true
sleep 2
python Run_Utils.py -r 1