prevent tapir rainbow demo from running out of host memory on free T4

PiperOrigin-RevId: 621204703 Change-Id: I5e021acd8090696984c074bed1eeae1477468844
google-deepmind · Apr 2, 2024 · b0c6aa6 · b0c6aa6
1 parent f6143e1
commit b0c6aa6
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/colabs/tapir_rainbow_demo.ipynb b/colabs/tapir_rainbow_demo.ipynb
@@ -205,9 +205,9 @@
       "source": [
         "# @title Inference function {form-width: \"25%\"}\n",
         "\n",
-        "resize_height = 512  # @param {type: \"integer\"}\n",
-        "resize_width = 512  # @param {type: \"integer\"}\n",
-        "stride = 16  # @param {type: \"integer\"}\n",
+        "resize_height = 256  # @param {type: \"integer\"}\n",
+        "resize_width = 256  # @param {type: \"integer\"}\n",
+        "stride = 8  # @param {type: \"integer\"}\n",
         "query_frame = 0  # @param {type: \"integer\"}\n",
         "\n",
         "frames = media.resize_video(orig_frames, (resize_height, resize_width))\n",