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()
|