Minor modifications

yzhangcs · yzhangcs · commit ca0e7cf657e6 · 2020-11-27T13:44:39.000+08:00
diff --git a/supar/parsers/constituency.py b/supar/parsers/constituency.py
@@ -274,10 +274,12 @@ def build(cls, path,
             'eos_index': WORD.eos_index,
             'feat_pad_index': FEAT.pad_index
         })
+        logger.info(f"{transform}")
 
         logger.info("Building the model")
-        model = cls.MODEL(**args)
-        model.load_pretrained(WORD.embed).to(args.device)
+        model = cls.MODEL(**args).load_pretrained(WORD.embed).to(args.device)
+        logger.info(f"{model}\n")
+
         optimizer = Adam(model.parameters(), **optimizer_args)
         scheduler = ExponentialLR(optimizer, **scheduler_args)
 
diff --git a/supar/parsers/dependency.py b/supar/parsers/dependency.py
@@ -279,10 +279,11 @@ def build(cls, path,
             'bos_index': WORD.bos_index,
             'feat_pad_index': FEAT.pad_index,
         })
+        logger.info(f"{transform}")
 
         logger.info("Building the model")
-        model = cls.MODEL(**args)
-        model.load_pretrained(WORD.embed).to(args.device)
+        model = cls.MODEL(**args).load_pretrained(WORD.embed).to(args.device)
+        logger.info(f"{model}\n")
 
         optimizer = Adam(model.parameters(), **optimizer_args)
         scheduler = ExponentialLR(optimizer, **scheduler_args)
@@ -923,10 +924,11 @@ def build(cls, path,
             'bos_index': WORD.bos_index,
             'feat_pad_index': FEAT.pad_index
         })
+        logger.info(f"{transform}")
 
         logger.info("Building the model")
-        model = cls.MODEL(**args)
-        model = model.load_pretrained(WORD.embed).to(args.device)
+        model = cls.MODEL(**args).load_pretrained(WORD.embed).to(args.device)
+        logger.info(f"{model}\n")
 
         optimizer = Adam(model.parameters(), **optimizer_args)
         scheduler = ExponentialLR(optimizer, **scheduler_args)
