3a181c8d229d3f45d6457cd329d2336b07b2330b,autokeras/pretrained/face_detector.py,FaceDetector,__init__,#FaceDetector#,276

Before Change

    def __init__(self):
        super(FaceDetector, self).__init__()
        pnet, rnet, onet = self.load()
        self.device = get_device()

        self.pnet_detector = PNet()
        if torch.cuda.is_available():
            self.pnet_detector.load_state_dict(torch.load(pnet))

After Change

    def __init__(self, **kwargs):
        super().__init__(**kwargs)
        pnet, rnet, onet = self.local_paths[0], self.local_paths[1], self.local_paths[2]

        self.pnet_detector = PNet()
        if torch.cuda.is_available():

Instances