(self,
train_dataset,
test_dataset,
model,
adv_model,
adv_dataset_dir,
loss,
y_target,
poisoned_rate,
adv_transform=Compose([transforms.ToTensor()]),
pattern=None,
weight=None,
eps=8,
alpha=1.5,
steps=100,
max_pixel=255,
poisoned_transform_train_index=0,
poisoned_transform_test_index=0,
poisoned_target_transform_index=0,
schedule=None,
seed=0,
deterministic=False)
| 479 | """ |
| 480 | |
| 481 | def __init__(self, |
| 482 | train_dataset, |
| 483 | test_dataset, |
| 484 | model, |
| 485 | adv_model, |
| 486 | adv_dataset_dir, |
| 487 | loss, |
| 488 | y_target, |
| 489 | poisoned_rate, |
| 490 | adv_transform=Compose([transforms.ToTensor()]), |
| 491 | pattern=None, |
| 492 | weight=None, |
| 493 | eps=8, |
| 494 | alpha=1.5, |
| 495 | steps=100, |
| 496 | max_pixel=255, |
| 497 | poisoned_transform_train_index=0, |
| 498 | poisoned_transform_test_index=0, |
| 499 | poisoned_target_transform_index=0, |
| 500 | schedule=None, |
| 501 | seed=0, |
| 502 | deterministic=False): |
| 503 | |
| 504 | super(LabelConsistent, self).__init__( |
| 505 | train_dataset=train_dataset, |
| 506 | test_dataset=test_dataset, |
| 507 | model=model, |
| 508 | loss=loss, |
| 509 | schedule=schedule, |
| 510 | seed=seed, |
| 511 | deterministic=deterministic) |
| 512 | |
| 513 | if poisoned_rate > 0: |
| 514 | self.whole_adv_dataset, self.target_adv_dataset, poisoned_set = self._get_adv_dataset( |
| 515 | train_dataset, |
| 516 | adv_model=adv_model, |
| 517 | adv_dataset_dir=adv_dataset_dir, |
| 518 | adv_transform=adv_transform, |
| 519 | eps=eps/max_pixel, |
| 520 | alpha=alpha/max_pixel, |
| 521 | steps=steps, |
| 522 | y_target=y_target, |
| 523 | poisoned_rate=poisoned_rate) |
| 524 | |
| 525 | self.poisoned_train_dataset = CreatePoisonedTargetDataset( |
| 526 | self.target_adv_dataset, |
| 527 | poisoned_set, |
| 528 | pattern, |
| 529 | weight, |
| 530 | poisoned_transform_train_index) |
| 531 | else: |
| 532 | self.poisoned_train_dataset = train_dataset |
| 533 | |
| 534 | self.poisoned_test_dataset = CreatePoisonedDataset( |
| 535 | test_dataset, |
| 536 | y_target, |
| 537 | 1.0, |
| 538 | pattern, |
nothing calls this directly
no test coverage detected