def _save_image_graphs(args, all_net, all_init_net): print('Saving model graph...') mutils.save_graph( all_net.Proto(), os.path.join(args.out_dir, "model_def.png"), op_only=False) print('Model def image saved to {}.'.format(args.out_dir))