summaryrefslogtreecommitdiff
path: root/realtime/mkconfig.py
diff options
context:
space:
mode:
Diffstat (limited to 'realtime/mkconfig.py')
-rwxr-xr-xrealtime/mkconfig.py11
1 files changed, 8 insertions, 3 deletions
diff --git a/realtime/mkconfig.py b/realtime/mkconfig.py
index 71944080..f9938051 100755
--- a/realtime/mkconfig.py
+++ b/realtime/mkconfig.py
@@ -59,9 +59,14 @@ def main():
shutil.copy(weights_final, os.path.join(output_d, 'weights.final'))
# other options
- # TODO: automatically set some things here
- with open(os.path.join(output_d, 'rt.ini'), 'w') as rt_ini:
- pass
+ rt_ini = os.path.join(output_d, 'rt.ini')
+ with open(rt_ini, 'w') as out:
+ if libcdec_ff_hpyplm_so and corpus_hpyplm:
+ out.write('hpyplm=true\n')
+ else:
+ out.write('hpyplm=false\n')
+ out.write('metric=ibm_bleu\n')
+ sys.stderr.write('IMPORTANT: add any additional options such as metric=meteor to {}\n'.format(rt_ini))
if __name__ == '__main__':
main()