def _call_cntk(self, config_file_name, config_content, action_name): ''' Calls the CNTK executable on the `config_content`. Args: config_file_name (str): the name of the configuration file config_content (str): a string containing the configuration action_name (str): the name of the action in cntk configuration file Returns: the output generated by the CNTK executable, which is used to retrieve the node shapes. ''' filename = self._save_file(config_file_name, config_content, action_name) try: output_bytes = subprocess.check_output( [get_cntk_cmd(), 'configFile=%s' % filename], stderr=subprocess.STDOUT) output = output_bytes.decode('utf-8') with open(os.path.join(self.directory, 'cntk.log'), 'w') as log: log.write(output) except subprocess.CalledProcessError as e: with open('error.txt', 'w') as f: f.write(e.output.decode('utf-8')) print("=" * 50) print(e.output.decode('utf-8')) print("=" * 50) raise if not output: raise ValueError('no output returned') return output
def _call_cntk(self, config_file_name, config_content, action_name): ''' Calls the CNTK executable on the `config_content`. Args: config_file_name (str): the name of the configuration file config_content (str): a string containing the configuration action_name (str): the name of the action in cntk configuration file Returns: the output generated by the CNTK executable, which is used to retrieve the node shapes. ''' filename = self._save_file(config_file_name, config_content, action_name) try: output_bytes = subprocess.check_output( [get_cntk_cmd(), 'configFile=%s' % filename], stderr=subprocess.STDOUT) output = output_bytes.decode('utf-8') with open(os.path.join(self.directory, 'cntk.log'), 'w') as log: log.write(output) except subprocess.CalledProcessError as e: with open('error.txt', 'w') as f: f.write(e.output.decode('utf-8')) print("=" * 50) print(e.output.decode('utf-8')) print("=" * 50) raise if not output: raise ValueError('no output returned') return output
def _call_cntk(self, config_file_name, config_content): ''' Calls the CNTK exe :param config_file_name: the name of the configuration file :param config_content: a string containing the configuration Returns the output generated by the CNTK executable, which is used to retrieve the node shapes. ''' filename = os.path.join(self.directory, config_file_name) with open(os.path.join(self.directory, filename), 'w') as out: out.write(config_content) try: output_bytes = subprocess.check_output( [get_cntk_cmd(), 'configFile=%s' % filename], stderr=subprocess.STDOUT) output = output_bytes.decode('utf-8') with open(os.path.join(self.directory, 'cntk.log'), 'w') as log: log.write(output) except subprocess.CalledProcessError as e: print(e.output.decode('utf-8'), file=open('error.txt', 'w')) raise if not output: raise ValueError('no output returned') return output