online_attacks/scripts/eval_classifiers.py [28:51]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
args, _ = parser.parse_known_args()

if args.dataset == DatasetType.MNIST:
    parser.add_argument(
        "--source_model_type",
        default=MnistModel.MODEL_A,
        type=MnistModel,
        choices=MnistModel,
    )
    parser.add_argument(
        "--target_model_type", default=None, type=MnistModel, choices=MnistModel
    )
elif args.dataset == DatasetType.CIFAR:
    parser.add_argument(
        "--source_model_type",
        default=CifarModel.VGG_16,
        type=CifarModel,
        choices=CifarModel,
    )
    parser.add_argument(
        "--target_model_type", default=None, type=CifarModel, choices=CifarModel
    )

args = parser.parse_args()
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



online_attacks/scripts/hist_classifiers.py [96:119]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    args, _ = parser.parse_known_args()

    if args.dataset == DatasetType.MNIST:
        parser.add_argument(
            "--source_model_type",
            default=MnistModel.MODEL_A,
            type=MnistModel,
            choices=MnistModel,
        )
        parser.add_argument(
            "--target_model_type", default=None, type=MnistModel, choices=MnistModel
        )
    elif args.dataset == DatasetType.CIFAR:
        parser.add_argument(
            "--source_model_type",
            default=CifarModel.VGG_16,
            type=CifarModel,
            choices=CifarModel,
        )
        parser.add_argument(
            "--target_model_type", default=None, type=CifarModel, choices=CifarModel
        )

    args = parser.parse_args()
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -