model = load_model(
    basic_model_path,
    model_type="badnet",
    input_channels=train_data_loader.dataset.channels,
    output_num=train_data_loader.dataset.class_num,
    device=device,
)