next · 9c2cb063be - disrupting-deepfakes - MS-GitHub-Backup (Gitea)

CalvinBackup/disrupting-deepfakes

next

This commit is contained in:

Nataniel Ruiz

2019-12-24 14:07:25 -04:00

parent f352bfd25d

commit 9c2cb063be

2 changed files with 5 additions and 3 deletions

									
										stargan/attacks.py
									
		+1
		-1
	
												View File
												
				@@ -35,7 +35,7 @@ class LinfPGDAttack(object):

				                y = torch.FloatTensor(y).to(self.device)

				            self.model.zero_grad()

				            loss = self.loss_fn(output, y)

				            loss = -self.loss_fn(output, y)

				            loss.backward()

				            grad = X.grad