(args)
| 222 | return args |
| 223 | |
| 224 | def main(args): |
| 225 | if args.config_args: |
| 226 | config = json.loads(_jsonnet.evaluate_file(args.config, tla_codes={'args': args.config_args})) |
| 227 | else: |
| 228 | config = json.loads(_jsonnet.evaluate_file(args.config)) |
| 229 | |
| 230 | if 'model_name' in config: |
| 231 | args.logdir = os.path.join(args.logdir, config['model_name']) |
| 232 | |
| 233 | output_path = args.output.replace('__LOGDIR__', args.logdir) |
| 234 | if os.path.exists(output_path): |
| 235 | print('Output file {} already exists'.format(output_path)) |
| 236 | sys.exit(1) |
| 237 | |
| 238 | inferer = Inferer(config) |
| 239 | model = inferer.load_model(args.logdir, args.step) |
| 240 | inferer.infer(model, output_path, args) |
| 241 | |
| 242 | |
| 243 | if __name__ == '__main__': |
no test coverage detected