def main()

in vision/run_weak_strong.py [0:0]


def main(
    batch_size: int = 128,
    weak_model_name: str = "alexnet",
    strong_model_name: str = "resnet50_dino",
    n_train: int = 40000,
    seed: int = 0,
    data_path: str = "/root/",
    n_epochs: int = 10,
    lr: float = 1e-3,