def display_model(opt) -> Tuple[str, str, str]:
"""
Run display_model.py.
:return: (stdout_train, stdout_valid, stdout_test)
"""
import parlai.scripts.display_model as dm
parser = dm.setup_args()
parser.set_params(**opt)
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)
