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

View File

@ -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':