0f67d39
1
2
3
4
5
6
7
8
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, )
