original_docstring: NesT small version with sequence length 196 for 224x224 inputs.
docstring: NesT small version with sequence length 196 for 224x224 inputs.
docstring_tokens: NesT small version with sequence length 196 for 224x224 inputs .
code: def nest_small_s196_224(config):
    """NesT small version with sequence length 196 for 224x224 inputs."""
    nest = default_config()
    nest.init_patch_embed_size = 4
    nest.patch_size = 14
    nest.num_layers_per_block = [2, 2, 20]
    nest.embedding_dim = 96
    nest.num_heads = 3
    nest.scale_hidden_dims = 2
    nest.stochastic_depth_drop = 0.3
    nest.attn_type = "local_multi_head"
  
    if config.get("nest"):
      nest.update(config.nest)
    return functools.partial(NestNet, config=nest)
code_tokens: def nest_small_s196_224 ( config ) : nest = default_config ( ) nest . init_patch_embed_size = 4 nest . patch_size = 14 nest . num_layers_per_block = [ 2 , 2 , 20 ] nest . embedding_dim = 96 nest . num_heads = 3 nest . scale_hidden_dims = 2 nest . stochastic_depth_drop = 0.3 nest . attn_type = "local_multi_head" if config . get ( "nest" ) : nest . update ( config . nest ) return functools . partial ( NestNet , config = nest )
short_docstring: NesT small version with sequence length 196 for 224x224 inputs.
short_docstring_tokens: NesT small version with sequence length 196 for 224x224 inputs .
comment: """NesT small version with sequence length 196 for 224x224 inputs."""