local

Author: schenglee

examples/pytorch/semantic_parsing/graph2tree/jobs/src/runner.py

@@ -138,6 +138,8 @@ def _build_dataloader(self):
     def _build_model(self):
         '''For encoder-decoder'''
         self.model = Graph2Tree.from_args(self.opt, vocab_model=self.vocab_model)
+        for name, param in self.model.named_parameters():
+            print(name)
         # self.model.init(self.opt["init_weight"])
         self.model.to(self.device)