(self, timer_flag=False)
if network == "mobile0.25":
            cfg = cfg_mnet
        elif network == "resnet50":
            cfg = cfg_re50
if network == "mobile0.25":
            cfg = cfg_mnet
        elif network == "resnet50":
            cfg = cfg_re50
if network == mobile0 . 25 : cfg = cfg_mnet elif network == resnet50 : cfg = cfg_re50
def __init__(self, timer_flag=False):
        torch.set_grad_enabled(False)
        '''
        if network == "mobile0.25":
            cfg = cfg_mnet
        elif network == "resnet50":
            cfg = cfg_re50
        '''
        self.cfg = cfg_mnet    
        # net and model
        net = RetinaFace(cfg=self.cfg, phase = 'test')
        self.net = load_model(net, trained_model, cpu)
        self.net.eval()
        #print('Finished loading model!')
        #print(net)
        #cudnn.benchmark = True
        self.device = torch.device("cpu" if cpu else "cuda")
        net = net.to(self.device)
        self.timer_flag = timer_flag
def __init__ ( self , timer_flag = False ) : torch . set_grad_enabled ( False ) self . cfg = cfg_mnet # net and model net = RetinaFace ( cfg = self . cfg , phase = 'test' ) self . net = load_model ( net , trained_model , cpu ) self . net . eval ( ) #print('Finished loading model!') #print(net) #cudnn.benchmark = True self . device = torch . device ( "cpu" if cpu else "cuda" ) net = net . to ( self . device ) self . timer_flag = timer_flag