9 lines
265 B
Python
9 lines
265 B
Python
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() |