"...git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "2cdac665b0706d3e6a60f98123283734d8d98aeb"
Unverified Commit d2d88226 authored by Matt's avatar Matt Committed by GitHub
Browse files

TF SAM memory reduction (#23732)

* Extremely small change to TF SAM dummies to reduce memory usage on build

* remove debug breakpoint

* Debug print statement to track array sizes

* More debug shape printing

* More debug shape printing

* Now remove the debug shape printing

* make fixup

* make fixup
parent 28aa438c
...@@ -1155,7 +1155,7 @@ class TFSamPreTrainedModel(TFPreTrainedModel): ...@@ -1155,7 +1155,7 @@ class TFSamPreTrainedModel(TFPreTrainedModel):
""" """
VISION_DUMMY_INPUTS = tf.random.uniform( VISION_DUMMY_INPUTS = tf.random.uniform(
shape=( shape=(
3, 1,
self.config.vision_config.num_channels, self.config.vision_config.num_channels,
self.config.vision_config.image_size, self.config.vision_config.image_size,
self.config.vision_config.image_size, self.config.vision_config.image_size,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment