ankile / adversarial-diffusion Goto Github PK

Thank you for the work!

I found an missing term purified in your code, could you please modify this part~
Thank you very much!

`preds_adv = np.array([])
preds_pur = np.array([])
ys = np.array([])
y_counter = 0

for i in range(1000):

Adv, X, y = purified[i]
X = X.to("cuda")
Adv = Adv.to("cuda")
pred_pur = cnn(X)
pred_adv = cnn(Adv)
preds_adv = np.append(preds_adv,np.argmax(pred_adv.detach().cpu().numpy(),axis=1))
preds_pur = np.append(preds_pur,np.argmax(pred_pur.detach().cpu().numpy(),axis=1))
y_counter+=y.numpy()
ys = np.append(ys, y.numpy())`