diff --git a/predict.py b/predict.py index 2e16dfb5..ec36986d 100644 --- a/predict.py +++ b/predict.py @@ -213,6 +213,10 @@ def get_args(): setattr(args, r, config[r]) elif 'cove' in r: setattr(args, r, False) + elif 'elmo' in r: + setattr(args, r, False) + elif 'glove_and_char' in r: + setattr(args, r, True) else: setattr(args, r, None) args.dropout_ratio = 0.0