import torch.optim as optim import torchvision from torchvision import transforms, models, datasets import imageio import time import warnings import random import sys import copy import json from PIL import Image
data_d