CoolFace
Apppublic

DIaac/FGSM_attack_demo

sourceHugging Faceupdated 2y agoView on Hugging Face
0likes
rootutilsbase_env.py
base_env.py4 linesDownload Raw Back to utils
1import torch
2
3device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
4