def __init__( self, config, data_loader, val_data_loader=None, test_data_loader=None, ): Trainer.__init__(self, config, data_loader, val_data_loader) self.best_val_metric = 'hit_ratio' self.train_max_iter = config.train_max_iter self.sift = cv2.xfeatures2d.SIFT_create()
def __init__(self, args): Trainer.__init__(self, args)