Skip to content

Commit

Permalink
Merge pull request #44 from togethercomputer/arsh/eng-12852-document-…
Browse files Browse the repository at this point in the history
…the-warmup-ratio-argument-in-docs-and-the-api

Add warmup_ratio to finetune API reference
  • Loading branch information
azahed98 authored Oct 20, 2024
2 parents 6d5cca2 + 520c947 commit 356f9cd
Showing 1 changed file with 7 additions and 0 deletions.
7 changes: 7 additions & 0 deletions openapi.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -400,6 +400,11 @@ paths:
format: float
default: 0.00001
description: Learning rate multiplier to use for training
warmup_ratio:
type: number
format: float
default: 0.0
description: The percent of steps at the start of training to linearly increase the learning-rate.
suffix:
type: string
description: Suffix that will be added to your fine-tuned model name
Expand Down Expand Up @@ -1731,6 +1736,8 @@ components:
type: integer
learning_rate:
type: number
warmup_ratio:
type: number
eval_steps:
type: integer
training_type:
Expand Down

0 comments on commit 356f9cd

Please sign in to comment.