Read configs from INI file

zysite · zysite · commit e50f1f39f1dc · 2019-05-02T14:14:28.000+08:00
diff --git a/config.py b/config.py
@@ -1,28 +1,37 @@
 # -*- coding: utf-8 -*-
 
+from ast import literal_eval
+from configparser import ConfigParser
+
 
 class Config(object):
 
-    # [Network]
-    n_embed = 100
-    n_tag_embed = 100
-    embed_dropout = 0.33
-    n_lstm_hidden = 400
-    n_lstm_layers = 3
-    lstm_dropout = 0.33
-    n_mlp_arc = 500
-    n_mlp_rel = 100
-    mlp_dropout = 0.33
-
-    # [Optimizer]
-    lr = 2e-3
-    beta_1 = 0.9
-    beta_2 = 0.9
-    epsilon = 1e-12
-    decay = .75
-    decay_steps = 5000
-
-    # [Run]
-    batch_size = 200
-    epochs = 1000
-    patience = 100
+    def __init__(self, fname):
+        super(Config, self).__init__()
+
+        self.config = ConfigParser()
+        self.config.read(fname)
+        self.kwargs = dict((option, literal_eval(value))
+                           for section in self.config.sections()
+                           for option, value in self.config.items(section))
+
+    def __repr__(self):
+        info = f"{self.__class__.__name__}:\n"
+        for i, (option, value) in enumerate(self.kwargs.items()):
+            info += f"{option:15} {value:<25}" + ('\n' if i % 2 > 0 else '')
+        if i % 2 == 0:
+            info += '\n'
+
+        return info
+
+    def __getattr__(self, attr):
+        return self.kwargs.get(attr, None)
+
+    def __getstate__(self):
+        return vars(self)
+
+    def __setstate__(self, state):
+        self.__dict__.update(state)
+
+    def update(self, kwargs):
+        self.kwargs.update(kwargs)
diff --git a/run.py b/run.py
@@ -6,28 +6,33 @@
 
 import torch
 
+from config import Config
+
+
 if __name__ == '__main__':
     parser = argparse.ArgumentParser(
         description='Create the Biaffine Parser model.'
     )
-    subparsers = parser.add_subparsers(title='Commands')
+    subparsers = parser.add_subparsers(title='Commands', dest='mode')
     subcommands = {
         'evaluate': Evaluate(),
         'predict': Predict(),
         'train': Train()
     }
     for name, subcommand in subcommands.items():
         subparser = subcommand.add_subparser(name, subparsers)
+        subparser.add_argument('--conf', '-c', default='config.ini',
+                               help='path to config file')
+        subparser.add_argument('--model', '-m', default='exp/ptb/model.tag',
+                               help='path to model file')
+        subparser.add_argument('--vocab', '-v', default='exp/ptb/vocab.tag',
+                               help='path to vocab file')
         subparser.add_argument('--device', '-d', default='-1',
                                help='ID of GPU to use')
         subparser.add_argument('--seed', '-s', default=1, type=int,
                                help='seed for generating random numbers')
         subparser.add_argument('--threads', '-t', default=4, type=int,
                                help='max num of threads')
-        subparser.add_argument('--file', '-f', default='model.pt',
-                               help='path to model file')
-        subparser.add_argument('--vocab', '-v', default='vocab.pt',
-                               help='path to vocabulary file')
     args = parser.parse_args()
 
     print(f"Set the max num of threads to {args.threads}")
@@ -37,4 +42,11 @@
     torch.manual_seed(args.seed)
     os.environ['CUDA_VISIBLE_DEVICES'] = args.device
 
-    args.func(args)
+    print(f"Override the default configs with parsed arguments")
+    config = Config(args.conf)
+    config.update(vars(args))
+    print(config)
+
+    print(f"Run the subcommand in mode {args.mode}")
+    cmd = subcommands[args.mode]
+    cmd(config)
