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