online_attacks/scripts/batch_online_attacks.py [35:51]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    parser.add_argument("--slurm", type=str, default="")
    if args.dataset == DatasetType.MNIST:
        parser.add_argument(
            "--model_type",
            nargs="+",
            default=MnistModel,
            type=MnistModel,
            choices=MnistModel,
        )
    elif args.dataset == DatasetType.CIFAR:
        parser.add_argument(
            "--model_type",
            nargs="+",
            default=CifarModel,
            type=CifarModel,
            choices=CifarModel,
        )
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



online_attacks/scripts/compute_hist_all.py [111:127]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    parser.add_argument("--slurm", type=str, default="")
    if args.dataset == DatasetType.MNIST:
        parser.add_argument(
            "--model_type",
            nargs="+",
            default=MnistModel,
            type=MnistModel,
            choices=MnistModel,
        )
    elif args.dataset == DatasetType.CIFAR:
        parser.add_argument(
            "--model_type",
            nargs="+",
            default=CifarModel,
            type=CifarModel,
            choices=CifarModel,
        )
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -