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,