Skip to content

Commit 9f982e9

Browse files
xuanlinli17xuanlinli17
committed
Add a note in colab on turning off denoisers as colab would crash if the denoiser is turned on
1 parent 5922dc3 commit 9f982e9

File tree

1 file changed

+28
-24
lines changed

1 file changed

+28
-24
lines changed

‎example.ipynb‎

Lines changed: 28 additions & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -33,20 +33,20 @@
3333
},
3434
{
3535
"cell_type": "code",
36-
"execution_count": 10,
36+
"execution_count": 4,
3737
"metadata": {
3838
"colab": {
3939
"base_uri": "https://localhost:8080/"
4040
},
4141
"id": "T5aYcrQzH9qo",
42-
"outputId": "1adee762-5d14-4ffc-bac4-c4b45c6fea09"
42+
"outputId": "1223a1ca-ba09-43b5-d3aa-a9a66ab73362"
4343
},
4444
"outputs": [
4545
{
4646
"output_type": "stream",
4747
"name": "stdout",
4848
"text": [
49-
"Mon May 6 07:28:10 2024 \n",
49+
"Mon May 6 07:35:28 2024 \n",
5050
"+---------------------------------------------------------------------------------------+\n",
5151
"| NVIDIA-SMI 535.104.05 Driver Version: 535.104.05 CUDA Version: 12.2 |\n",
5252
"|-----------------------------------------+----------------------+----------------------+\n",
@@ -55,7 +55,7 @@
5555
"| | | MIG M. |\n",
5656
"|=========================================+======================+======================|\n",
5757
"| 0 Tesla T4 Off | 00000000:00:04.0 Off | 0 |\n",
58-
"| N/A 36C P8 9W / 70W | 0MiB / 15360MiB | 0% Default |\n",
58+
"| N/A 34C P8 9W / 70W | 0MiB / 15360MiB | 0% Default |\n",
5959
"| | | N/A |\n",
6060
"+-----------------------------------------+----------------------+----------------------+\n",
6161
" \n",
@@ -76,7 +76,7 @@
7676
},
7777
{
7878
"cell_type": "code",
79-
"execution_count": 11,
79+
"execution_count": 5,
8080
"metadata": {
8181
"id": "pI0Ge9viDPks"
8282
},
@@ -92,13 +92,13 @@
9292
},
9393
{
9494
"cell_type": "code",
95-
"execution_count": 12,
95+
"execution_count": 6,
9696
"metadata": {
9797
"colab": {
9898
"base_uri": "https://localhost:8080/"
9999
},
100100
"id": "r_EBPnGtU1zA",
101-
"outputId": "6b82e4dc-ed45-4a61-8bab-d211a0d7394f"
101+
"outputId": "aaa3d71c-9c7d-4398-bedf-2035e87b6535"
102102
},
103103
"outputs": [
104104
{
@@ -122,7 +122,7 @@
122122
},
123123
{
124124
"cell_type": "code",
125-
"execution_count": 13,
125+
"execution_count": null,
126126
"metadata": {
127127
"colab": {
128128
"base_uri": "https://localhost:8080/"
@@ -241,7 +241,7 @@
241241
},
242242
{
243243
"cell_type": "code",
244-
"execution_count": 14,
244+
"execution_count": null,
245245
"metadata": {
246246
"colab": {
247247
"base_uri": "https://localhost:8080/"
@@ -288,7 +288,7 @@
288288
},
289289
{
290290
"cell_type": "code",
291-
"execution_count": 15,
291+
"execution_count": null,
292292
"metadata": {
293293
"colab": {
294294
"base_uri": "https://localhost:8080/"
@@ -359,7 +359,7 @@
359359
},
360360
{
361361
"cell_type": "code",
362-
"execution_count": 16,
362+
"execution_count": null,
363363
"metadata": {
364364
"colab": {
365365
"base_uri": "https://localhost:8080/"
@@ -384,7 +384,7 @@
384384
},
385385
{
386386
"cell_type": "code",
387-
"execution_count": 17,
387+
"execution_count": null,
388388
"metadata": {
389389
"id": "4stLX8cdaWD3"
390390
},
@@ -408,11 +408,10 @@
408408
},
409409
{
410410
"cell_type": "code",
411-
"execution_count": 18,
411+
"execution_count": null,
412412
"metadata": {
413413
"colab": {
414-
"base_uri": "https://localhost:8080/",
415-
"height": 0
414+
"base_uri": "https://localhost:8080/"
416415
},
417416
"id": "pMhxVs4nC9NT",
418417
"outputId": "876a4a87-f265-468d-806b-dd15c9d54895"
@@ -499,9 +498,8 @@
499498
},
500499
{
501500
"cell_type": "code",
502-
"execution_count": 19,
501+
"execution_count": null,
503502
"metadata": {
504-
"cellView": "form",
505503
"id": "XgxjPOtxoAw6"
506504
},
507505
"outputs": [],
@@ -538,9 +536,8 @@
538536
},
539537
{
540538
"cell_type": "code",
541-
"execution_count": 24,
539+
"execution_count": null,
542540
"metadata": {
543-
"cellView": "form",
544541
"colab": {
545542
"base_uri": "https://localhost:8080/"
546543
},
@@ -568,7 +565,12 @@
568565
" env.close()\n",
569566
" del env\n",
570567
"env = simpler_env.make(task_name)\n",
568+
"\n",
569+
"# Note: we turned off the denoiser as the colab kernel will crash if it's turned on\n",
570+
"# To use the denoiser, please git clone our SIMPLER environments\n",
571+
"# and perform evaluations locally.\n",
571572
"sapien.render_config.rt_use_denoiser = False\n",
573+
"\n",
572574
"obs, reset_info = env.reset()\n",
573575
"instruction = env.get_language_instruction()\n",
574576
"print(\"Reset info\", reset_info)\n",
@@ -582,9 +584,8 @@
582584
},
583585
{
584586
"cell_type": "code",
585-
"execution_count": 25,
587+
"execution_count": null,
586588
"metadata": {
587-
"cellView": "form",
588589
"colab": {
589590
"base_uri": "https://localhost:8080/"
590591
},
@@ -779,7 +780,7 @@
779780
},
780781
{
781782
"cell_type": "code",
782-
"execution_count": 26,
783+
"execution_count": null,
783784
"metadata": {
784785
"cellView": "form",
785786
"colab": {
@@ -863,7 +864,7 @@
863864
},
864865
{
865866
"cell_type": "code",
866-
"execution_count": 27,
867+
"execution_count": null,
867868
"metadata": {
868869
"colab": {
869870
"base_uri": "https://localhost:8080/",
@@ -941,7 +942,10 @@
941942
"source": [
942943
"# @markdown RT-1-X close drawer\n",
943944
"print(task_name, model_name)\n",
944-
"mediapy.show_video(images, fps=10)"
945+
"mediapy.show_video(images, fps=10)\n",
946+
"# Note: we turned off the denoiser as the colab kernel will crash if it's turned on\n",
947+
"# To use the denoiser, please git clone our SIMPLER environments\n",
948+
"# and perform evaluations locally."
945949
]
946950
},
947951
{

0 commit comments

Comments
 (0)