Amend
This commit is contained in:
@@ -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
14
train.sh
Executable 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
|
||||||
Reference in New Issue
Block a user