def display_model(opt) -> Tuple[str, str]: """ Run display_model.py. :return: (stdout_valid, stdout_test) """ import parlai.scripts.display_model as dm parser = dm.setup_args() parser.set_params(**opt) popt = parser.parse_args(print_args=False) with capture_output() as train_output: # evalmode so that we don't hit train_step popt['datatype'] = 'train:evalmode:stream' dm.display_model(popt) with capture_output() as valid_output: popt['datatype'] = 'valid:stream' dm.display_model(popt) with capture_output() as test_output: popt['datatype'] = 'test:stream' dm.display_model(popt) return (train_output.getvalue(), valid_output.getvalue(), test_output.getvalue())
#!/usr/bin/env python3 # Copyright (c) 2017-present, Facebook, Inc. # All rights reserved. # This source code is licensed under the BSD-style license found in the # LICENSE file in the root directory of this source tree. An additional grant # of patent rights can be found in the PATENTS file in the same directory. """ See parlai.scripts.display_model for documentation. """ from parlai.scripts.display_model import display_model, setup_args if __name__ == '__main__': parser = setup_args() opt = parser.parse_args() display_model(opt)
======= <<<<<<< HEAD <<<<<<< HEAD popt = parser.parse_args([], print_args=False) ======= popt = parser.parse_args(print_args=False) >>>>>>> 4f6b99642d60aff1a41b9eae8bd2ccd9e40ebba4 >>>>>>> origin/master ======= popt = parser.parse_args(print_args=False) >>>>>>> 4f6b99642d60aff1a41b9eae8bd2ccd9e40ebba4 >>>>>>> ef574cebef2a8d5aa38b73176b1e71a919d6670f with capture_output() as train_output: # evalmode so that we don't hit train_step popt['datatype'] = 'train:evalmode:stream' dm.display_model(popt) with capture_output() as valid_output: popt['datatype'] = 'valid:stream' dm.display_model(popt) with capture_output() as test_output: popt['datatype'] = 'test:stream' dm.display_model(popt) return (train_output.getvalue(), valid_output.getvalue(), test_output.getvalue()) <<<<<<< HEAD def download_unittest_models(): """ Download the unittest pretrained models. """ from parlai.core.params import ParlaiParser