liwenyun
/
GanAttack
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update pkg

This commit is contained in:
Li Wenyun 2023-12-12 17:55:35 +08:00
parent 76c26f6f9d
commit 4a74e4337e
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
utils.py
View File

@ -4,6 +4,7 @@ import torch.optim as optim
from PIL import Image from PIL import Image
from torchvision import models from torchvision import models
import numpy as np import numpy as np
import PIL
def get_model(config): def get_model(config):
if config.dataset == 'gender_dataset': if config.dataset == 'gender_dataset':