From 4a74e4337ebd813352357bb58e7f5fb154e28e8c Mon Sep 17 00:00:00 2001 From: Li Wenyun Date: Tue, 12 Dec 2023 17:55:35 +0800 Subject: [PATCH] update pkg --- utils.py | 1 + 1 file changed, 1 insertion(+) diff --git a/utils.py b/utils.py index 2acd663..ab18a6f 100644 --- a/utils.py +++ b/utils.py @@ -4,6 +4,7 @@ import torch.optim as optim from PIL import Image from torchvision import models import numpy as np +import PIL def get_model(config): if config.dataset == 'gender_dataset':