import efficientnetv2_m 

import torch 

from flyai.utilsimport remote_helper

path = remote_helper.get_remote_data('https://www.flyai.com/m/pre_efficientnetv2-m.pth')   

self.net = efficientnetv2_m(num_classes) 

self.net.to(device) 

weights_dict = torch.load(path, map_location=device) 

load_weights_dict = {k: v for k, v in weights_dict.items() if self.net.state_dict()[k].numel() == v.numel()}

self.net.load_state_dict(load_weights_dict, strict=False)