import torch
import torchvision
from torchvision.models import EfficientNet_B2_Weights , efficientnet_b2

efficient_weight = EfficientNet_B2_Weights.DEFAULT

efficient_transformer = efficient_weight.transforms()

efficient_model = torch.load("efficient_b2.pth").to("cpu")