(self, root_node)
| 903 | self.run_pretrain_routine(model) |
| 904 | |
| 905 | def resolve_root_node_address(self, root_node): |
| 906 | if '[' in root_node: |
| 907 | name = root_node.split('[')[0] |
| 908 | number = root_node.split(',')[0] |
| 909 | if '-' in number: |
| 910 | number = number.split('-')[0] |
| 911 | |
| 912 | number = re.sub('[^0-9]', '', number) |
| 913 | root_node = name + number |
| 914 | |
| 915 | return root_node |
| 916 | |
| 917 | def log_metrics(self, metrics, grad_norm_dic, step=None): |
| 918 | """Logs the metric dict passed in. |