MohammadAliMKH's picture
Update model.py
ac12ce6
raw
history blame
No virus
304 Bytes
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_model_101.pth" , map_location=torch.device('cpu'))