From b0c6aa6fdfe72c71b379764f9bfde8088221f30f Mon Sep 17 00:00:00 2001 From: Carl Doersch Date: Tue, 2 Apr 2024 09:32:51 -0700 Subject: [PATCH] prevent tapir rainbow demo from running out of host memory on free T4 PiperOrigin-RevId: 621204703 Change-Id: I5e021acd8090696984c074bed1eeae1477468844 --- colabs/tapir_rainbow_demo.ipynb | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/colabs/tapir_rainbow_demo.ipynb b/colabs/tapir_rainbow_demo.ipynb index 4f6bc8f..8d4114b 100644 --- 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",