Skip to content

Commit

Permalink
reduced maximum number of frozen layers
Browse files Browse the repository at this point in the history
  • Loading branch information
MaxTeselkin committed Jan 7, 2025
1 parent 157210b commit 3ff5d76
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion train/src/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -525,7 +525,7 @@ def update_globals(new_dataset_ids):
"fine-tuning model on small datasets"
),
)
n_frozen_layers_input = InputNumber(value=1, min=1, max=200)
n_frozen_layers_input = InputNumber(value=1, min=1, max=90)
n_frozen_layers_input_f = Field(
content=n_frozen_layers_input, title="Number of layers to freeze"
)
Expand Down

0 comments on commit 3ff5d76

Please sign in to comment.