bad-net / temp
mhyatt000's picture
deploy
0f67d39
raw
history blame
195 Bytes
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,
)