def main(args):
# initialize dataset
audio_dataset = AudioDataset(args.manifest,
batch_size=args.batch_size,
root_dir=args.root_dir)
audio_dataloader = audio_dataset.dataloader()
attack = Attack(args.model_path,
batch_size=args.batch_size,
lr_stage1=args.lr_stage1,
lr_stage2=args.lr_stage2,
num_iter_stage1=args.num_iter_stage1,
num_iter_stage2=args.num_iter_stage2)
# initialize attack class
attack.attack_stage1(audio_dataloader)
for idx, batch in enumerate(audio_dataloader):
#print(idx, batch)
for key, value in batch.items():
try:
print(key, ':', value.shape)
except AttributeError:
print(key, ':', value)
exit()
