targetのテンソル形状の不一致

---------------------------------------------------------------------------
IndexError                                Traceback (most recent call last)
/tmp/ipykernel_3445/4182338255.py in <module>
     54 
     55 # Generate adversarial examples
---> 56 df_adv_lpf = gen_adv(config, 'LowProFool')
     57 df_adv_df = gen_adv(config, 'Deepfool')
     58 config['AdvData'] = {'LowProFool' : df_adv_lpf, 'Deepfool' : df_adv_df}

/tmp/ipykernel_3445/4270127548.py in gen_adv(config, method)
     41 
     42         if method == 'LowProFool':
---> 43             orig_pred, adv_pred, x_adv, loop_i = lowProFool(x_tensor, model, weights, encoded_bounds, maxiters, alpha, lambda_)
     44         elif method == 'Deepfool':
     45             orig_pred, adv_pred, x_adv, loop_i = deepfool(x_tensor, model, maxiters, alpha, encoded_bounds, weights=[])

/src/Adverse.py in lowProFool(x, model, weights, bounds, maxiters, alpha, lambda_)
     71 
     72     target = torch.zeros_like(output)
---> 73     target[:, target_pred] = 1 # scatterの代わり
     74     # target = Variable(target, requires_grad=False)
     75 

IndexError: index 1 is out of bounds for dimension 0 with size 1
ShotaArima / demo-lowprofool

targetのテンソル形状の不一致 #11
