Steve Nyemba 3 years ago
parent 4345146f3a
commit 7a22314a46

@ -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)
Loading…
Cancel
Save