refactor(cli): migrate from argparse to typer for command-line interface

This commit is contained in:
2026-03-06 11:41:35 +08:00
parent 33f2b8f542
commit 7dbd704d6b
8 changed files with 195 additions and 91 deletions

View File

@@ -0,0 +1,20 @@
import typer
def train(
ctx: typer.Context,
epoch_size: int = typer.Option(10, "--epoch", "-e", help="Number of epochs"),
batch_size: int = typer.Option(64, "--batch", "-b", help="Batch size"),
lr: float = typer.Option(1e-4, "--lr", "-l", help="Learning rate"),
checkpoint_path: str = typer.Option(
"hash_checkpoint.pt", "--checkpoint", "-c", help="Checkpoint path"
),
):
from compressors import train as train_module
train_module(
epoch_size=epoch_size,
batch_size=batch_size,
lr=lr,
checkpoint_path=checkpoint_path,
)