Spaces:

DIaac
/

FGSM_attack_demo

Build error

Upload folder using huggingface_hub

8cd44cd verified over 2 years ago

87 Bytes

import torch

device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')

	import torch

	device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')