From c9308d7e5eaf7e49012531565920587fd580fd56 Mon Sep 17 00:00:00 2001 From: maryam-zia Date: Thu, 30 Oct 2025 15:25:47 -0400 Subject: [PATCH 01/13] Change colab ubuntu version. --- .github/workflows/colab.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/colab.yml b/.github/workflows/colab.yml index 7946adfd5d..4ece61ca6a 100644 --- a/.github/workflows/colab.yml +++ b/.github/workflows/colab.yml @@ -16,7 +16,7 @@ on: jobs: colab: - runs-on: ubuntu-22.04 + runs-on: ubuntu-latest env: COLAB_ALWAYS_INSTALL_XVFB: 1 QLEARNING_NUM_TRAINING_STEPS: 5 From bd3c61988fa2501a2f121fa1a7ca0716bc1b51c0 Mon Sep 17 00:00:00 2001 From: maryam-zia Date: Thu, 30 Oct 2025 15:59:11 -0400 Subject: [PATCH 02/13] Add timeout parameter. --- .github/workflows/colab.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/colab.yml b/.github/workflows/colab.yml index 4ece61ca6a..e3ce570986 100644 --- a/.github/workflows/colab.yml +++ b/.github/workflows/colab.yml @@ -54,7 +54,7 @@ jobs: python -m pip install --progress-bar=off -e ./ml-agents python -m pip install --progress-bar=off -r colab_requirements.txt - name: Execute notebook - run: jupyter nbconvert --to notebook --execute --log-level=DEBUG --ExecutePreprocessor.kernel_name=python3 --output output-${{ matrix.notebook_path }} colab/${{ matrix.notebook_path }} + run: jupyter nbconvert --to notebook --execute --log-level=DEBUG --ExecutePreprocessor.kernel_name=python3 --ExecutePreprocessor.timeout=600 --output output-${{ matrix.notebook_path }} colab/${{ matrix.notebook_path }} - name: Upload colab results uses: actions/upload-artifact@v4 with: From 88f53945c3bc6760f3fe01ca565bf703cf3f9b9b Mon Sep 17 00:00:00 2001 From: maryam-zia Date: Thu, 30 Oct 2025 16:25:32 -0400 Subject: [PATCH 03/13] Move timeout to colab --- .github/workflows/colab.yml | 4 ++-- colab/Colab_UnityEnvironment_1_Run.ipynb | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/.github/workflows/colab.yml b/.github/workflows/colab.yml index e3ce570986..7946adfd5d 100644 --- a/.github/workflows/colab.yml +++ b/.github/workflows/colab.yml @@ -16,7 +16,7 @@ on: jobs: colab: - runs-on: ubuntu-latest + runs-on: ubuntu-22.04 env: COLAB_ALWAYS_INSTALL_XVFB: 1 QLEARNING_NUM_TRAINING_STEPS: 5 @@ -54,7 +54,7 @@ jobs: python -m pip install --progress-bar=off -e ./ml-agents python -m pip install --progress-bar=off -r colab_requirements.txt - name: Execute notebook - run: jupyter nbconvert --to notebook --execute --log-level=DEBUG --ExecutePreprocessor.kernel_name=python3 --ExecutePreprocessor.timeout=600 --output output-${{ matrix.notebook_path }} colab/${{ matrix.notebook_path }} + run: jupyter nbconvert --to notebook --execute --log-level=DEBUG --ExecutePreprocessor.kernel_name=python3 --output output-${{ matrix.notebook_path }} colab/${{ matrix.notebook_path }} - name: Upload colab results uses: actions/upload-artifact@v4 with: diff --git a/colab/Colab_UnityEnvironment_1_Run.ipynb b/colab/Colab_UnityEnvironment_1_Run.ipynb index e70dca7b8a..471a598abf 100644 --- a/colab/Colab_UnityEnvironment_1_Run.ipynb +++ b/colab/Colab_UnityEnvironment_1_Run.ipynb @@ -217,7 +217,7 @@ "\n", "from mlagents_envs.registry import default_registry\n", "\n", - "env = default_registry[env_id].make()" + "env = default_registry[env_id].make(timeout_wait=120)" ], "execution_count": 3, "outputs": [ From ef463fb394b8a87380a9ae11fd8ab70f416b587b Mon Sep 17 00:00:00 2001 From: maryam-zia Date: Thu, 30 Oct 2025 16:36:24 -0400 Subject: [PATCH 04/13] Bump to latest --- .github/workflows/colab.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/colab.yml b/.github/workflows/colab.yml index 7946adfd5d..4ece61ca6a 100644 --- a/.github/workflows/colab.yml +++ b/.github/workflows/colab.yml @@ -16,7 +16,7 @@ on: jobs: colab: - runs-on: ubuntu-22.04 + runs-on: ubuntu-latest env: COLAB_ALWAYS_INSTALL_XVFB: 1 QLEARNING_NUM_TRAINING_STEPS: 5 From 9c63f168c5f1b6b4ac8f02c68aa878b6c83db5e0 Mon Sep 17 00:00:00 2001 From: maryam-zia Date: Thu, 30 Oct 2025 16:45:52 -0400 Subject: [PATCH 05/13] Remove upper bound limit --- .github/workflows/colab.yml | 2 +- colab/Colab_UnityEnvironment_1_Run.ipynb | 2 +- ml-agents/setup.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/.github/workflows/colab.yml b/.github/workflows/colab.yml index 4ece61ca6a..7946adfd5d 100644 --- a/.github/workflows/colab.yml +++ b/.github/workflows/colab.yml @@ -16,7 +16,7 @@ on: jobs: colab: - runs-on: ubuntu-latest + runs-on: ubuntu-22.04 env: COLAB_ALWAYS_INSTALL_XVFB: 1 QLEARNING_NUM_TRAINING_STEPS: 5 diff --git a/colab/Colab_UnityEnvironment_1_Run.ipynb b/colab/Colab_UnityEnvironment_1_Run.ipynb index 471a598abf..e70dca7b8a 100644 --- a/colab/Colab_UnityEnvironment_1_Run.ipynb +++ b/colab/Colab_UnityEnvironment_1_Run.ipynb @@ -217,7 +217,7 @@ "\n", "from mlagents_envs.registry import default_registry\n", "\n", - "env = default_registry[env_id].make(timeout_wait=120)" + "env = default_registry[env_id].make()" ], "execution_count": 3, "outputs": [ diff --git a/ml-agents/setup.py b/ml-agents/setup.py index 2aec4e2cd6..16be25d8b3 100644 --- a/ml-agents/setup.py +++ b/ml-agents/setup.py @@ -62,7 +62,7 @@ def run(self): "Pillow>=4.2.1", "protobuf>=3.6,<3.21", "pyyaml>=3.1.0", - "torch>=2.1.1,<=2.8.0", + "torch>=2.1.1", "tensorboard>=2.14", # adding six explicit dependency since tensorboard needs it but doesn't declare it as a dep "six>=1.16", From fc5fea4b82aaa1939cca20cb27eca0bdbe2e8d68 Mon Sep 17 00:00:00 2001 From: maryam-zia Date: Thu, 30 Oct 2025 17:24:23 -0400 Subject: [PATCH 06/13] Test --- .github/workflows/colab.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/colab.yml b/.github/workflows/colab.yml index 7946adfd5d..7e26a39798 100644 --- a/.github/workflows/colab.yml +++ b/.github/workflows/colab.yml @@ -54,7 +54,7 @@ jobs: python -m pip install --progress-bar=off -e ./ml-agents python -m pip install --progress-bar=off -r colab_requirements.txt - name: Execute notebook - run: jupyter nbconvert --to notebook --execute --log-level=DEBUG --ExecutePreprocessor.kernel_name=python3 --output output-${{ matrix.notebook_path }} colab/${{ matrix.notebook_path }} + run: jupyter nbconvert --to notebook --execute --log-level=DEBUG --ExecutePreprocessor.kernel_name=python3 --ExecutePreprocessor.timeout=120 --output output-${{ matrix.notebook_path }} colab/${{ matrix.notebook_path }} - name: Upload colab results uses: actions/upload-artifact@v4 with: From f1c7d05d8bfa532bd76a69fba8b6a559e414ac88 Mon Sep 17 00:00:00 2001 From: maryam-zia Date: Thu, 30 Oct 2025 17:34:11 -0400 Subject: [PATCH 07/13] Change versions --- .github/workflows/colab.yml | 2 +- ml-agents/setup.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/colab.yml b/.github/workflows/colab.yml index 7e26a39798..f8b5af3d6d 100644 --- a/.github/workflows/colab.yml +++ b/.github/workflows/colab.yml @@ -16,7 +16,7 @@ on: jobs: colab: - runs-on: ubuntu-22.04 + runs-on: ubuntu-latest env: COLAB_ALWAYS_INSTALL_XVFB: 1 QLEARNING_NUM_TRAINING_STEPS: 5 diff --git a/ml-agents/setup.py b/ml-agents/setup.py index 16be25d8b3..2aec4e2cd6 100644 --- a/ml-agents/setup.py +++ b/ml-agents/setup.py @@ -62,7 +62,7 @@ def run(self): "Pillow>=4.2.1", "protobuf>=3.6,<3.21", "pyyaml>=3.1.0", - "torch>=2.1.1", + "torch>=2.1.1,<=2.8.0", "tensorboard>=2.14", # adding six explicit dependency since tensorboard needs it but doesn't declare it as a dep "six>=1.16", From d48d4a25f20c4e7708aa4271aca8f1ad991d54bc Mon Sep 17 00:00:00 2001 From: maryam-zia Date: Thu, 30 Oct 2025 17:49:21 -0400 Subject: [PATCH 08/13] Test --- colab/Colab_UnityEnvironment_1_Run.ipynb | 55 ++---------------------- 1 file changed, 3 insertions(+), 52 deletions(-) diff --git a/colab/Colab_UnityEnvironment_1_Run.ipynb b/colab/Colab_UnityEnvironment_1_Run.ipynb index e70dca7b8a..4db8560620 100644 --- a/colab/Colab_UnityEnvironment_1_Run.ipynb +++ b/colab/Colab_UnityEnvironment_1_Run.ipynb @@ -206,58 +206,9 @@ "end_time": "2023-10-04T12:52:26.115543Z" } }, - "source": [ - "# -----------------\n", - "# This code is used to close an env that might not have been closed before\n", - "try:\n", - " env.close()\n", - "except:\n", - " pass\n", - "# -----------------\n", - "\n", - "from mlagents_envs.registry import default_registry\n", - "\n", - "env = default_registry[env_id].make()" - ], - "execution_count": 3, - "outputs": [ - { - "name": "stdout", - "output_type": "stream", - "text": [ - "[UnityMemory] Configuration Parameters - Can be set up in boot.config\n", - " \"memorysetup-bucket-allocator-granularity=16\"\n", - " \"memorysetup-bucket-allocator-bucket-count=8\"\n", - " \"memorysetup-bucket-allocator-block-size=4194304\"\n", - " \"memorysetup-bucket-allocator-block-count=1\"\n", - " \"memorysetup-main-allocator-block-size=16777216\"\n", - " \"memorysetup-thread-allocator-block-size=16777216\"\n", - " \"memorysetup-gfx-main-allocator-block-size=16777216\"\n", - " \"memorysetup-gfx-thread-allocator-block-size=16777216\"\n", - " \"memorysetup-cache-allocator-block-size=4194304\"\n", - " \"memorysetup-typetree-allocator-block-size=2097152\"\n", - " \"memorysetup-profiler-bucket-allocator-granularity=16\"\n", - " \"memorysetup-profiler-bucket-allocator-bucket-count=8\"\n", - " \"memorysetup-profiler-bucket-allocator-block-size=4194304\"\n", - " \"memorysetup-profiler-bucket-allocator-block-count=1\"\n", - " \"memorysetup-profiler-allocator-block-size=16777216\"\n", - " \"memorysetup-profiler-editor-allocator-block-size=1048576\"\n", - " \"memorysetup-temp-allocator-size-main=4194304\"\n", - " \"memorysetup-job-temp-allocator-block-size=2097152\"\n", - " \"memorysetup-job-temp-allocator-block-size-background=1048576\"\n", - " \"memorysetup-job-temp-allocator-reduction-small-platforms=262144\"\n", - " \"memorysetup-allocator-temp-initial-block-size-main=262144\"\n", - " \"memorysetup-allocator-temp-initial-block-size-worker=262144\"\n", - " \"memorysetup-temp-allocator-size-background-worker=32768\"\n", - " \"memorysetup-temp-allocator-size-job-worker=262144\"\n", - " \"memorysetup-temp-allocator-size-preload-manager=262144\"\n", - " \"memorysetup-temp-allocator-size-nav-mesh-worker=65536\"\n", - " \"memorysetup-temp-allocator-size-audio-worker=65536\"\n", - " \"memorysetup-temp-allocator-size-cloud-worker=32768\"\n", - " \"memorysetup-temp-allocator-size-gfx=262144\"\n" - ] - } - ] + "source": "# -----------------\n# This code is used to close an env that might not have been closed before\ntry:\n env.close()\nexcept:\n pass\n# -----------------\n\nfrom mlagents_envs.registry import default_registry\n\nenv = default_registry[env_id].make(timeout_wait=120)", + "execution_count": null, + "outputs": [] }, { "cell_type": "markdown", From bf39b06951a18a0937f3ab10cdcc13239c3d42c1 Mon Sep 17 00:00:00 2001 From: maryam-zia Date: Thu, 30 Oct 2025 17:57:00 -0400 Subject: [PATCH 09/13] Increase cell timeout --- .github/workflows/colab.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/colab.yml b/.github/workflows/colab.yml index f8b5af3d6d..ad2c4a48f4 100644 --- a/.github/workflows/colab.yml +++ b/.github/workflows/colab.yml @@ -54,7 +54,7 @@ jobs: python -m pip install --progress-bar=off -e ./ml-agents python -m pip install --progress-bar=off -r colab_requirements.txt - name: Execute notebook - run: jupyter nbconvert --to notebook --execute --log-level=DEBUG --ExecutePreprocessor.kernel_name=python3 --ExecutePreprocessor.timeout=120 --output output-${{ matrix.notebook_path }} colab/${{ matrix.notebook_path }} + run: jupyter nbconvert --to notebook --execute --log-level=DEBUG --ExecutePreprocessor.kernel_name=python3 --ExecutePreprocessor.timeout=300 --output output-${{ matrix.notebook_path }} colab/${{ matrix.notebook_path }} - name: Upload colab results uses: actions/upload-artifact@v4 with: From 9bf77e4d03004e0220d7239739348a94f78bbf0e Mon Sep 17 00:00:00 2001 From: maryam-zia Date: Thu, 30 Oct 2025 18:06:56 -0400 Subject: [PATCH 10/13] Increase Unity timeout --- colab/Colab_UnityEnvironment_1_Run.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/colab/Colab_UnityEnvironment_1_Run.ipynb b/colab/Colab_UnityEnvironment_1_Run.ipynb index 4db8560620..3aa18bb4e7 100644 --- a/colab/Colab_UnityEnvironment_1_Run.ipynb +++ b/colab/Colab_UnityEnvironment_1_Run.ipynb @@ -206,7 +206,7 @@ "end_time": "2023-10-04T12:52:26.115543Z" } }, - "source": "# -----------------\n# This code is used to close an env that might not have been closed before\ntry:\n env.close()\nexcept:\n pass\n# -----------------\n\nfrom mlagents_envs.registry import default_registry\n\nenv = default_registry[env_id].make(timeout_wait=120)", + "source": "# -----------------\n# This code is used to close an env that might not have been closed before\ntry:\n env.close()\nexcept:\n pass\n# -----------------\n\nfrom mlagents_envs.registry import default_registry\n\nenv = default_registry[env_id].make(timeout_wait=180)", "execution_count": null, "outputs": [] }, From 8a3cf12f198357341c87eaf91fdce66d89f326ec Mon Sep 17 00:00:00 2001 From: maryam-zia Date: Thu, 30 Oct 2025 18:16:52 -0400 Subject: [PATCH 11/13] Set no_graphics to True --- colab/Colab_UnityEnvironment_1_Run.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/colab/Colab_UnityEnvironment_1_Run.ipynb b/colab/Colab_UnityEnvironment_1_Run.ipynb index 3aa18bb4e7..ec3aa10f01 100644 --- a/colab/Colab_UnityEnvironment_1_Run.ipynb +++ b/colab/Colab_UnityEnvironment_1_Run.ipynb @@ -206,7 +206,7 @@ "end_time": "2023-10-04T12:52:26.115543Z" } }, - "source": "# -----------------\n# This code is used to close an env that might not have been closed before\ntry:\n env.close()\nexcept:\n pass\n# -----------------\n\nfrom mlagents_envs.registry import default_registry\n\nenv = default_registry[env_id].make(timeout_wait=180)", + "source": "# -----------------\n# This code is used to close an env that might not have been closed before\ntry:\n env.close()\nexcept:\n pass\n# -----------------\n\nfrom mlagents_envs.registry import default_registry\n\nenv = default_registry[env_id].make( no_graphics=True)", "execution_count": null, "outputs": [] }, From 3a83fef37dd7fea6cdb264545ef8a8abca9c294a Mon Sep 17 00:00:00 2001 From: maryam-zia Date: Thu, 30 Oct 2025 18:26:50 -0400 Subject: [PATCH 12/13] Clean --- .github/workflows/colab.yml | 2 +- colab/Colab_UnityEnvironment_1_Run.ipynb | 55 ++++++++++++++++++++++-- 2 files changed, 53 insertions(+), 4 deletions(-) diff --git a/.github/workflows/colab.yml b/.github/workflows/colab.yml index ad2c4a48f4..4ece61ca6a 100644 --- a/.github/workflows/colab.yml +++ b/.github/workflows/colab.yml @@ -54,7 +54,7 @@ jobs: python -m pip install --progress-bar=off -e ./ml-agents python -m pip install --progress-bar=off -r colab_requirements.txt - name: Execute notebook - run: jupyter nbconvert --to notebook --execute --log-level=DEBUG --ExecutePreprocessor.kernel_name=python3 --ExecutePreprocessor.timeout=300 --output output-${{ matrix.notebook_path }} colab/${{ matrix.notebook_path }} + run: jupyter nbconvert --to notebook --execute --log-level=DEBUG --ExecutePreprocessor.kernel_name=python3 --output output-${{ matrix.notebook_path }} colab/${{ matrix.notebook_path }} - name: Upload colab results uses: actions/upload-artifact@v4 with: diff --git a/colab/Colab_UnityEnvironment_1_Run.ipynb b/colab/Colab_UnityEnvironment_1_Run.ipynb index ec3aa10f01..ded77696ad 100644 --- a/colab/Colab_UnityEnvironment_1_Run.ipynb +++ b/colab/Colab_UnityEnvironment_1_Run.ipynb @@ -206,9 +206,58 @@ "end_time": "2023-10-04T12:52:26.115543Z" } }, - "source": "# -----------------\n# This code is used to close an env that might not have been closed before\ntry:\n env.close()\nexcept:\n pass\n# -----------------\n\nfrom mlagents_envs.registry import default_registry\n\nenv = default_registry[env_id].make( no_graphics=True)", - "execution_count": null, - "outputs": [] + "source": [ + "# -----------------\n", + "# This code is used to close an env that might not have been closed before\n", + "try:\n", + " env.close()\n", + "except:\n", + " pass\n", + "# -----------------\n", + "\n", + "from mlagents_envs.registry import default_registry\n", + "\n", + "env = default_registry[env_id].make(no_graphics=True)" + ], + "execution_count": 3, + "outputs": [ + { + "name": "stdout", + "output_type": "stream", + "text": [ + "[UnityMemory] Configuration Parameters - Can be set up in boot.config\n", + " \"memorysetup-bucket-allocator-granularity=16\"\n", + " \"memorysetup-bucket-allocator-bucket-count=8\"\n", + " \"memorysetup-bucket-allocator-block-size=4194304\"\n", + " \"memorysetup-bucket-allocator-block-count=1\"\n", + " \"memorysetup-main-allocator-block-size=16777216\"\n", + " \"memorysetup-thread-allocator-block-size=16777216\"\n", + " \"memorysetup-gfx-main-allocator-block-size=16777216\"\n", + " \"memorysetup-gfx-thread-allocator-block-size=16777216\"\n", + " \"memorysetup-cache-allocator-block-size=4194304\"\n", + " \"memorysetup-typetree-allocator-block-size=2097152\"\n", + " \"memorysetup-profiler-bucket-allocator-granularity=16\"\n", + " \"memorysetup-profiler-bucket-allocator-bucket-count=8\"\n", + " \"memorysetup-profiler-bucket-allocator-block-size=4194304\"\n", + " \"memorysetup-profiler-bucket-allocator-block-count=1\"\n", + " \"memorysetup-profiler-allocator-block-size=16777216\"\n", + " \"memorysetup-profiler-editor-allocator-block-size=1048576\"\n", + " \"memorysetup-temp-allocator-size-main=4194304\"\n", + " \"memorysetup-job-temp-allocator-block-size=2097152\"\n", + " \"memorysetup-job-temp-allocator-block-size-background=1048576\"\n", + " \"memorysetup-job-temp-allocator-reduction-small-platforms=262144\"\n", + " \"memorysetup-allocator-temp-initial-block-size-main=262144\"\n", + " \"memorysetup-allocator-temp-initial-block-size-worker=262144\"\n", + " \"memorysetup-temp-allocator-size-background-worker=32768\"\n", + " \"memorysetup-temp-allocator-size-job-worker=262144\"\n", + " \"memorysetup-temp-allocator-size-preload-manager=262144\"\n", + " \"memorysetup-temp-allocator-size-nav-mesh-worker=65536\"\n", + " \"memorysetup-temp-allocator-size-audio-worker=65536\"\n", + " \"memorysetup-temp-allocator-size-cloud-worker=32768\"\n", + " \"memorysetup-temp-allocator-size-gfx=262144\"\n" + ] + } + ] }, { "cell_type": "markdown", From 1ad3d438dde8ca503842c971bab6cca2f6aa6950 Mon Sep 17 00:00:00 2001 From: maryam-zia Date: Thu, 30 Oct 2025 18:28:56 -0400 Subject: [PATCH 13/13] Clean up --- colab/Colab_UnityEnvironment_1_Run.ipynb | 102 +++++++++++------------ 1 file changed, 51 insertions(+), 51 deletions(-) diff --git a/colab/Colab_UnityEnvironment_1_Run.ipynb b/colab/Colab_UnityEnvironment_1_Run.ipynb index ded77696ad..9ac80f920f 100644 --- a/colab/Colab_UnityEnvironment_1_Run.ipynb +++ b/colab/Colab_UnityEnvironment_1_Run.ipynb @@ -207,57 +207,57 @@ } }, "source": [ - "# -----------------\n", - "# This code is used to close an env that might not have been closed before\n", - "try:\n", - " env.close()\n", - "except:\n", - " pass\n", - "# -----------------\n", - "\n", - "from mlagents_envs.registry import default_registry\n", - "\n", - "env = default_registry[env_id].make(no_graphics=True)" - ], - "execution_count": 3, - "outputs": [ - { - "name": "stdout", - "output_type": "stream", - "text": [ - "[UnityMemory] Configuration Parameters - Can be set up in boot.config\n", - " \"memorysetup-bucket-allocator-granularity=16\"\n", - " \"memorysetup-bucket-allocator-bucket-count=8\"\n", - " \"memorysetup-bucket-allocator-block-size=4194304\"\n", - " \"memorysetup-bucket-allocator-block-count=1\"\n", - " \"memorysetup-main-allocator-block-size=16777216\"\n", - " \"memorysetup-thread-allocator-block-size=16777216\"\n", - " \"memorysetup-gfx-main-allocator-block-size=16777216\"\n", - " \"memorysetup-gfx-thread-allocator-block-size=16777216\"\n", - " \"memorysetup-cache-allocator-block-size=4194304\"\n", - " \"memorysetup-typetree-allocator-block-size=2097152\"\n", - " \"memorysetup-profiler-bucket-allocator-granularity=16\"\n", - " \"memorysetup-profiler-bucket-allocator-bucket-count=8\"\n", - " \"memorysetup-profiler-bucket-allocator-block-size=4194304\"\n", - " \"memorysetup-profiler-bucket-allocator-block-count=1\"\n", - " \"memorysetup-profiler-allocator-block-size=16777216\"\n", - " \"memorysetup-profiler-editor-allocator-block-size=1048576\"\n", - " \"memorysetup-temp-allocator-size-main=4194304\"\n", - " \"memorysetup-job-temp-allocator-block-size=2097152\"\n", - " \"memorysetup-job-temp-allocator-block-size-background=1048576\"\n", - " \"memorysetup-job-temp-allocator-reduction-small-platforms=262144\"\n", - " \"memorysetup-allocator-temp-initial-block-size-main=262144\"\n", - " \"memorysetup-allocator-temp-initial-block-size-worker=262144\"\n", - " \"memorysetup-temp-allocator-size-background-worker=32768\"\n", - " \"memorysetup-temp-allocator-size-job-worker=262144\"\n", - " \"memorysetup-temp-allocator-size-preload-manager=262144\"\n", - " \"memorysetup-temp-allocator-size-nav-mesh-worker=65536\"\n", - " \"memorysetup-temp-allocator-size-audio-worker=65536\"\n", - " \"memorysetup-temp-allocator-size-cloud-worker=32768\"\n", - " \"memorysetup-temp-allocator-size-gfx=262144\"\n" - ] - } - ] + "# -----------------\n", + "# This code is used to close an env that might not have been closed before\n", + "try:\n", + " env.close()\n", + "except:\n", + " pass\n", + "# -----------------\n", + "\n", + "from mlagents_envs.registry import default_registry\n", + "\n", + "env = default_registry[env_id].make(no_graphics=True)" + ], + "execution_count": 3, + "outputs": [ + { + "name": "stdout", + "output_type": "stream", + "text": [ + "[UnityMemory] Configuration Parameters - Can be set up in boot.config\n", + " \"memorysetup-bucket-allocator-granularity=16\"\n", + " \"memorysetup-bucket-allocator-bucket-count=8\"\n", + " \"memorysetup-bucket-allocator-block-size=4194304\"\n", + " \"memorysetup-bucket-allocator-block-count=1\"\n", + " \"memorysetup-main-allocator-block-size=16777216\"\n", + " \"memorysetup-thread-allocator-block-size=16777216\"\n", + " \"memorysetup-gfx-main-allocator-block-size=16777216\"\n", + " \"memorysetup-gfx-thread-allocator-block-size=16777216\"\n", + " \"memorysetup-cache-allocator-block-size=4194304\"\n", + " \"memorysetup-typetree-allocator-block-size=2097152\"\n", + " \"memorysetup-profiler-bucket-allocator-granularity=16\"\n", + " \"memorysetup-profiler-bucket-allocator-bucket-count=8\"\n", + " \"memorysetup-profiler-bucket-allocator-block-size=4194304\"\n", + " \"memorysetup-profiler-bucket-allocator-block-count=1\"\n", + " \"memorysetup-profiler-allocator-block-size=16777216\"\n", + " \"memorysetup-profiler-editor-allocator-block-size=1048576\"\n", + " \"memorysetup-temp-allocator-size-main=4194304\"\n", + " \"memorysetup-job-temp-allocator-block-size=2097152\"\n", + " \"memorysetup-job-temp-allocator-block-size-background=1048576\"\n", + " \"memorysetup-job-temp-allocator-reduction-small-platforms=262144\"\n", + " \"memorysetup-allocator-temp-initial-block-size-main=262144\"\n", + " \"memorysetup-allocator-temp-initial-block-size-worker=262144\"\n", + " \"memorysetup-temp-allocator-size-background-worker=32768\"\n", + " \"memorysetup-temp-allocator-size-job-worker=262144\"\n", + " \"memorysetup-temp-allocator-size-preload-manager=262144\"\n", + " \"memorysetup-temp-allocator-size-nav-mesh-worker=65536\"\n", + " \"memorysetup-temp-allocator-size-audio-worker=65536\"\n", + " \"memorysetup-temp-allocator-size-cloud-worker=32768\"\n", + " \"memorysetup-temp-allocator-size-gfx=262144\"\n" + ] + } + ] }, { "cell_type": "markdown",