Skip to content

Commit

Permalink
No public description
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 707060914
  • Loading branch information
tensorflower-gardener committed Dec 17, 2024
1 parent ee3c993 commit 789b2b3
Showing 1 changed file with 6 additions and 2 deletions.
8 changes: 6 additions & 2 deletions official/projects/pix2seq/configs/pix2seq.py
Original file line number Diff line number Diff line change
Expand Up @@ -130,9 +130,13 @@ class Pix2Seq(hyperparams.Config):
use_cls_token: bool = False
shared_decoder_embedding: bool = True
decoder_output_bias: bool = True
# The shape of the input image. For example, [640, 640, 3] for RGB.
# If using multiple backbones, this input size is understood to be the same
# for all backbones. If you need a separate input size for each backbone,
# please implement this behavior.
input_size: List[int] = dataclasses.field(default_factory=list)
# Backbones for each image modality. If just using RGB, you should only set
# one backbone.
# Backbones for each image modality. The RGB backbone is always the first one.
# If just using RGB, you should only set one backbone.
backbones: List[BackboneConfig] = dataclasses.field(
default_factory=lambda: [ # pylint: disable=g-long-lambda
BackboneConfig(
Expand Down

0 comments on commit 789b2b3

Please sign in to comment.