Skip to content

Commit

Permalink
Add layerscale back
Browse files Browse the repository at this point in the history
  • Loading branch information
D-Roberts committed May 7, 2023
1 parent 3036260 commit 4a64767
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -133,7 +133,7 @@ def __init__(
drop_path_rate: float = 0.0,
num_meta3d_blocks: int = 1,
distillation: bool = True,
use_layer_scale: bool = False,
use_layer_scale: bool = True,
layer_scale_init_value: float = 1e-5,
hidden_act: str = "gelu",
initializer_range: float = 0.02,
Expand Down

0 comments on commit 4a64767

Please sign in to comment.