Skip to content

Commit

Permalink
update specifying gpu in sim
Browse files Browse the repository at this point in the history
  • Loading branch information
@minoring
minoring committed Nov 18, 2023
1 parent f09405f commit faee3a7
Show file tree
Hide file tree
Showing 3 changed files with 6 additions and 6 deletions.
4 changes: 2 additions & 2 deletions docs/_sources/tutorials/furniture_sim.rst.txt
100644 → 100755
View file
Original file line number Diff line number Diff line change
Expand Up @@ -168,10 +168,10 @@ We provide automated furniture assembly scripts. It currently supports only ``on

.. code:: bash
python -m furniture_bench.scripts.collect_data --furniture <furniture> --scripted --is-sim --out-data-path <path/to/output> --gpu-id <gpu_id> --num-demos <num_demos> --headless
python -m furniture_bench.scripts.collect_data --furniture <furniture> --scripted --is-sim --out-data-path <path/to/output> --compute-device-id <compute_device_id> --graphics-device-id <graphics_device_id> --num-demos <num_demos> --headless
# E.g., collect 100 demonstrations for one_leg assembly
python -m furniture_bench.scripts.collect_data --furniture one_leg --scripted --is-sim --out-data-path scripted_sim_demo --gpu-id 0 --num-demos 100 --headless
python -m furniture_bench.scripts.collect_data --furniture one_leg --scripted --is-sim --out-data-path scripted_sim_demo --compute-device-id 0 --graphics-device-id 0 --num-demos 100 --headless
Here are some additional flags for ``collect_data.py``:

Expand Down
4 changes: 2 additions & 2 deletions docs/tutorials/furniture_sim.html
100644 → 100755
View file
Original file line number Diff line number Diff line change
Expand Up @@ -582,10 +582,10 @@ <h2>Automated Assembly Script<a class="headerlink" href="#automated-assembly-scr
<a class="reference internal image-reference" href="../_images/assembly_script.gif"><img alt="Assembly script" src="../_images/assembly_script.gif" style="width: 50%;" /></a>
</figure>
<p><strong>Data Collection:</strong> Using our scripted policy, you can collect <code class="docutils literal notranslate"><span class="pre">num-demos</span></code> demonstrations. If you use Docker, make sure you mount the output data path to the Docker container before you collect data.</p>
<div class="highlight-bash notranslate"><div class="highlight"><pre><span></span>python<span class="w"> </span>-m<span class="w"> </span>furniture_bench.scripts.collect_data<span class="w"> </span>--furniture<span class="w"> </span>&lt;furniture&gt;<span class="w"> </span>--scripted<span class="w"> </span>--is-sim<span class="w"> </span>--out-data-path<span class="w"> </span>&lt;path/to/output&gt;<span class="w"> </span>--gpu-id<span class="w"> </span>&lt;gpu_id&gt;<span class="w"> </span>--num-demos<span class="w"> </span>&lt;num_demos&gt;<span class="w"> </span>--headless
<div class="highlight-bash notranslate"><div class="highlight"><pre><span></span>python<span class="w"> </span>-m<span class="w"> </span>furniture_bench.scripts.collect_data<span class="w"> </span>--furniture<span class="w"> </span>&lt;furniture&gt;<span class="w"> </span>--scripted<span class="w"> </span>--is-sim<span class="w"> </span>--out-data-path<span class="w"> </span>&lt;path/to/output&gt;<span class="w"> </span>--compute-device-id<span class="w"> </span>&lt;compute_device_id&gt;<span class="w"> </span>--graphics-device-id<span class="w"> </span>&lt;graphics_device_id&gt;<span class="w"> </span>--num-demos<span class="w"> </span>&lt;num_demos&gt;<span class="w"> </span>--headless

<span class="c1"># E.g., collect 100 demonstrations for one_leg assembly</span>
python<span class="w"> </span>-m<span class="w"> </span>furniture_bench.scripts.collect_data<span class="w"> </span>--furniture<span class="w"> </span>one_leg<span class="w"> </span>--scripted<span class="w"> </span>--is-sim<span class="w"> </span>--out-data-path<span class="w"> </span>scripted_sim_demo<span class="w"> </span>--gpu-id<span class="w"> </span><span class="m">0</span><span class="w"> </span>--num-demos<span class="w"> </span><span class="m">100</span><span class="w"> </span>--headless
python<span class="w"> </span>-m<span class="w"> </span>furniture_bench.scripts.collect_data<span class="w"> </span>--furniture<span class="w"> </span>one_leg<span class="w"> </span>--scripted<span class="w"> </span>--is-sim<span class="w"> </span>--out-data-path<span class="w"> </span>scripted_sim_demo<span class="w"> </span>--compute-device-id<span class="w"> </span><span class="m">0</span><span class="w"> </span>--graphics-device-id<span class="w"> </span><span class="m">0</span><span class="w"> </span>--num-demos<span class="w"> </span><span class="m">100</span><span class="w"> </span>--headless
</pre></div>
</div>
<p>Here are some additional flags for <code class="docutils literal notranslate"><span class="pre">collect_data.py</span></code>:</p>
Expand Down
4 changes: 2 additions & 2 deletions docs_source/tutorials/furniture_sim.rst
100644 → 100755
View file
Original file line number Diff line number Diff line change
Expand Up @@ -168,10 +168,10 @@ We provide automated furniture assembly scripts. It currently supports only ``on

.. code:: bash
python -m furniture_bench.scripts.collect_data --furniture <furniture> --scripted --is-sim --out-data-path <path/to/output> --gpu-id <gpu_id> --num-demos <num_demos> --headless
python -m furniture_bench.scripts.collect_data --furniture <furniture> --scripted --is-sim --out-data-path <path/to/output> --compute-device-id <compute_device_id> --graphics-device-id <graphics_device_id> --num-demos <num_demos> --headless
# E.g., collect 100 demonstrations for one_leg assembly
python -m furniture_bench.scripts.collect_data --furniture one_leg --scripted --is-sim --out-data-path scripted_sim_demo --gpu-id 0 --num-demos 100 --headless
python -m furniture_bench.scripts.collect_data --furniture one_leg --scripted --is-sim --out-data-path scripted_sim_demo --compute-device-id 0 --graphics-device-id 0 --num-demos 100 --headless
Here are some additional flags for ``collect_data.py``:

Expand Down

0 comments on commit faee3a7

Please sign in to comment.