riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 33 Packages Projects Releases Wiki Activity

TF SAM memory reduction (#23732)

Browse Source 
* 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
This commit is contained in:
Matt 2023-05-24 15:59:02 +01:00 committed by GitHub
parent 28aa438cd2
commit d2d8822604
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/models/sam/modeling_tf_sam.py
View File

@ -1155,7 +1155,7 @@ class TFSamPreTrainedModel(TFPreTrainedModel):
"""
VISION_DUMMY_INPUTS = tf.random.uniform(
shape=(
3,
1,
self.config.vision_config.num_channels,
self.config.vision_config.image_size,
self.config.vision_config.image_size,