from models.eval.style import StyleTransferEvaluator def create_evaluator(model, opt_eval, env): type = opt_eval['type'] if type == 'style_transfer': return StyleTransferEvaluator(model, opt_eval, env) else: raise NotImplementedError()