| # Copyright (c) Meta Platforms, Inc. and affiliates. | |
| # All rights reserved. | |
| # | |
| # This source code is licensed under the license found in the | |
| # LICENSE file in the root directory of this source tree. | |
| import argparse | |
| from mmengine import Config, DictAction | |
| def parse_args(): | |
| parser = argparse.ArgumentParser(description='Print the whole config') | |
| parser.add_argument('config', help='config file path') | |
| parser.add_argument( | |
| '--options', nargs='+', action=DictAction, help='arguments in dict') | |
| args = parser.parse_args() | |
| return args | |
| def main(): | |
| args = parse_args() | |
| cfg = Config.fromfile(args.config) | |
| if args.options is not None: | |
| cfg.merge_from_dict(args.options) | |
| print(f'Config:\n{cfg.pretty_text}') | |
| if __name__ == '__main__': | |
| main() | |