Verified Commit 0e441c43 authored by Jakob Moser's avatar Jakob Moser
Browse files

Clarify variable name

parent ff0cb6ec

src/orangensaft/train/main.py

+2 −2
Original line number Diff line number Diff line
@@ -8,7 +8,7 @@ logger = logging.getLogger(__name__) 


@app.command()

def train(

    model: Annotated[

    model_name: Annotated[

        Literal["bert", "bart", "t5", "gpt2"],

        typer.Argument(help="The model to train."),

    ],
@@ -23,5 +23,5 @@ def train( 
    """

    Pre-train a Transformer model from scratch and store the weights on disk.

    """

    logging.info(f"Starting training for {model}, query/key relationship: {query_key_relationship}, seed: {seed}")

    logging.info(f"Starting training for {model_name}, query/key relationship: {query_key_relationship}, seed: {seed}")

    pass  # TODO Implement