| 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("/demo/efficient_b2.pth").to("cpu") | |