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
