Skip to content

Commit

Permalink
remove metrics from constructor
Browse files Browse the repository at this point in the history
  • Loading branch information
rifqiyan committed Mar 28, 2024
1 parent 5b0aa90 commit b917faa
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/axolotl/utils/callbacks/vessl_.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@

class VesslLogMetricsCallback(TrainerCallback):

def __init__(self, credential_path: str, metrics: List[str]) -> None:
def __init__(self, credential_path: str) -> None:
vessl.configure(credentials_file=credential_path)

def on_log(self, args: TrainingArguments, state: TrainerState, control: TrainerControl, logs: Dict[str, float] = None, **kwargs):
Expand Down

0 comments on commit b917faa

Please sign in to comment.