online_attacks/scripts/batch_online_attacks.py [18:33]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
if __name__ == "__main__":
    # Initialize submitit

    parser = argparse.ArgumentParser()
    parser.add_argument(
        "--dataset", default=DatasetType.MNIST, type=DatasetType, choices=DatasetType
    )
    parser.add_argument(
        "--attacker_type", default=Attacker.FGSM_ATTACK, type=Attacker, choices=Attacker
    )
    parser.add_argument("--batch_size", default=1000, type=int)
    parser.add_argument("--name", default="default", type=str)

    # Hack to be able to parse either MnistModel or CifarModel and to add slurm config
    args, _ = parser.parse_known_args()
    params = OmegaConf.structured(Params(**vars(args)))
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



online_attacks/scripts/random_eval.py [87:99]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    parser.add_argument(
        "--dataset", default=DatasetType.MNIST, type=DatasetType, choices=DatasetType
    )
    parser.add_argument(
        "--attacker_type", default=Attacker.FGSM_ATTACK, type=Attacker, choices=Attacker
    )
    parser.add_argument("--batch_size", default=1000, type=int)
    parser.add_argument("--name", default="default", type=str)
    args, _ = parser.parse_known_args()

    params = OmegaConf.structured(Params(**vars(args)))
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -