|
|
|
@ -336,9 +336,10 @@ class factory :
|
|
|
|
|
:param autopilot will generate output automatically
|
|
|
|
|
:param batch (default 2k) size of the batch
|
|
|
|
|
"""
|
|
|
|
|
if 'apply' not in _args :
|
|
|
|
|
return Trainer(**_args)
|
|
|
|
|
elif _args['apply'] == 'shuffe' :
|
|
|
|
|
|
|
|
|
|
if _args['apply'] == 'shuffe' :
|
|
|
|
|
return Shuffle(**_args)
|
|
|
|
|
elif _args['apply'] == 'generate' :
|
|
|
|
|
return Generator(**_args)
|
|
|
|
|
else:
|
|
|
|
|
return Trainer(**_args)
|