From patchwork Mon Apr 26 10:48:54 2021 Content-Type: text/plain; charset="utf-8" MIME-Version: 1.0 Content-Transfer-Encoding: 7bit X-Patchwork-Submitter: Lance Wang X-Patchwork-Id: 27423 Delivered-To: ffmpegpatchwork2@gmail.com Received: by 2002:a6b:5014:0:0:0:0:0 with SMTP id e20csp3850953iob; Mon, 26 Apr 2021 03:49:51 -0700 (PDT) X-Google-Smtp-Source: ABdhPJySgQBsVrUaXVBRuiT0SI1FVRo2LSlOksm6vXpLF6D5X9nW55Q/V62cMaF4t7053bovKXD1 X-Received: by 2002:a05:6402:4308:: with SMTP id m8mr20459083edc.209.1619434191008; Mon, 26 Apr 2021 03:49:51 -0700 (PDT) ARC-Seal: i=1; a=rsa-sha256; t=1619434191; cv=none; d=google.com; s=arc-20160816; b=r46NMI+8sEGHkwpSXxboVRSY9I7qqjBnar3niOLy1ZCTH+QiKRLD3hmMmYhcjQfLp+ VblNd3HObcjPHhqNVfDz947zjQGG/XkO8ZR9v5gDYk1RzK+fWYV3Gm0nAjWalGSBl0Ec /lK6waHEyVDZqBNab+sRrunE/y7mCz94vagdMZh7SoyZjcNJ6NHX/hWtKNTZmMB/15Op VQ3cNglTqUKmLWJp7OsUVZ4fLSU4IfRRR13O+DiCtO23GbElp2OgCodbgVAasRO5N3lR mcBDK8aRf1jRmYNZBaUg1nXjqkPcx6kzl0/sdrEmUwQksBHKgBTYx+JNetILsEFqW6ag q2rw== ARC-Message-Signature: i=1; a=rsa-sha256; c=relaxed/relaxed; d=google.com; s=arc-20160816; h=sender:errors-to:content-transfer-encoding:mime-version:cc:reply-to :list-subscribe:list-help:list-post:list-archive:list-unsubscribe :list-id:precedence:subject:references:in-reply-to:message-id:date :to:from:dkim-signature:delivered-to; bh=sZwQ+6SsHFSF/205WwjWFRSmHsvUh8i+GtBg9RH7WoQ=; b=wBoX3IBAlTe+Vf7Nx2hmSzQ2n3KlWVWERtHTNCk43cevRhithAVn/rA1fulvDY4ay6 H97nlupQtaOKFYUA+i+2zbA+Be59Najls968cdRkjtxEyVkqcCfDGPep6HoAruXaGuyY nBF5mqfPH5y35qzsMfAWTsmfFALZv643Xyklt3Nfkyf8kvR8FLcdu4SaZ3nliDDiXKhU Xxc2OCGxB99q1vIijZmLgHFfYswDaGw2YW+IOtEdIp4rQV6Y+NUNoplLJP5R1xDYzjI3 j6vWdXbSA/M4eN32Kiv2QQh5eWHw6XsX0cppKQzqE8pknoQzmaXbxSFrBtoV37QWIIu0 ebZQ== ARC-Authentication-Results: i=1; mx.google.com; dkim=neutral (body hash did not verify) header.i=@gmail.com header.s=20161025 header.b=M70StwlJ; spf=pass (google.com: domain of ffmpeg-devel-bounces@ffmpeg.org designates 79.124.17.100 as permitted sender) smtp.mailfrom=ffmpeg-devel-bounces@ffmpeg.org; dmarc=fail (p=NONE sp=QUARANTINE dis=NONE) header.from=gmail.com Return-Path: Received: from ffbox0-bg.mplayerhq.hu (ffbox0-bg.ffmpeg.org. [79.124.17.100]) by mx.google.com with ESMTP id k18si13055177edv.534.2021.04.26.03.49.50; Mon, 26 Apr 2021 03:49:50 -0700 (PDT) Received-SPF: pass (google.com: domain of ffmpeg-devel-bounces@ffmpeg.org designates 79.124.17.100 as permitted sender) client-ip=79.124.17.100; Authentication-Results: mx.google.com; dkim=neutral (body hash did not verify) header.i=@gmail.com header.s=20161025 header.b=M70StwlJ; spf=pass (google.com: domain of ffmpeg-devel-bounces@ffmpeg.org designates 79.124.17.100 as permitted sender) smtp.mailfrom=ffmpeg-devel-bounces@ffmpeg.org; dmarc=fail (p=NONE sp=QUARANTINE dis=NONE) header.from=gmail.com Received: from [127.0.1.1] (localhost [127.0.0.1]) by ffbox0-bg.mplayerhq.hu (Postfix) with ESMTP id 0AB9B6818DB; Mon, 26 Apr 2021 13:49:20 +0300 (EEST) X-Original-To: ffmpeg-devel@ffmpeg.org Delivered-To: ffmpeg-devel@ffmpeg.org Received: from mail-pl1-f172.google.com (mail-pl1-f172.google.com [209.85.214.172]) by ffbox0-bg.mplayerhq.hu (Postfix) with ESMTPS id 0E310689B7A for ; Mon, 26 Apr 2021 13:49:13 +0300 (EEST) Received: by mail-pl1-f172.google.com with SMTP id p17so2819796plf.12 for ; Mon, 26 Apr 2021 03:49:12 -0700 (PDT) DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=gmail.com; s=20161025; h=from:to:cc:subject:date:message-id:in-reply-to:references; bh=ioFrw0eAyj0lgL0cnNAZG1dHPU8wfASZZK/is0CN/mw=; b=M70StwlJ6w+aanb7W8cgYZO2enhsblxAH6nv+fCSk3s12EUNYcYze/V7/TgEo/A6r5 FeX0I9n7iFcx988x2EoBWd9CEvarAiE3abg2jhAiWswosJInIg2dXovPW0niuiBsAbld yNj1Y65wFmS6xZW/sJ8YC0XrAui0Fc44BnPoU4jxgmfqcENc5TGkbDj/OXx4bDzxLeVu k5Acl5GV3xdZksLkuZrZQsBkC8gUXYwNTeLryZ/6jcO0rxTTX0+m3FNrSSM0zv9dlOET GQtbYAJrz9nOBeNtAZoe3Hyzn2zL7f3xrpQYe0dnYouEFGnLsVsxKKW1j+bJXMrou95w hVTg== X-Google-DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=1e100.net; s=20161025; h=x-gm-message-state:from:to:cc:subject:date:message-id:in-reply-to :references; bh=ioFrw0eAyj0lgL0cnNAZG1dHPU8wfASZZK/is0CN/mw=; b=NQf+Xzj035hIAQEToBUJulRPDIv97osGyofSXY791/TlXm1B6vUsJmDan8u1Ia8HXu z335f7vjIdHYqKo9G6tI3TIYI/uSxhBeAvP26UgdBPFiq4Fyr8mL79itDGQrMzo8xsBH jYVyaX5MDdTrMHMH7EYVbTncX9DBMhnKa5dImpPRPZhFezOwU3Sx71KXarH7Bwe+6qMo 5Hxo+i8XRZcqpqpKfVnoONhjI5QaPsoe8CPQXY8P99a6cGRfWocZEMYn39Xtz3QLINJG JnP/kNaKW4au2C9mIMM6nupqupT07yGV8Y1xBs1GdgyF5Hh4Z8tfcbQbQc6CrAoQaE+6 skUw== X-Gm-Message-State: AOAM533Qlq5FI59k/aYS+kMWgGLV6JLezJL0jbk6pWLPwcUVvjuS/bDl z9i/3a3Vh5fsT8YIjMqNjSMWh8nWSt8= X-Received: by 2002:a17:902:f212:b029:ec:af98:8350 with SMTP id m18-20020a170902f212b02900ecaf988350mr18010868plc.37.1619434151437; Mon, 26 Apr 2021 03:49:11 -0700 (PDT) Received: from vpn2.localdomain ([161.117.202.209]) by smtp.gmail.com with ESMTPSA id u14sm1275592pjy.6.2021.04.26.03.49.10 (version=TLS1_2 cipher=ECDHE-ECDSA-AES128-GCM-SHA256 bits=128/128); Mon, 26 Apr 2021 03:49:11 -0700 (PDT) From: lance.lmwang@gmail.com To: ffmpeg-devel@ffmpeg.org Date: Mon, 26 Apr 2021 18:48:54 +0800 Message-Id: <1619434136-17760-4-git-send-email-lance.lmwang@gmail.com> X-Mailer: git-send-email 1.8.3.1 In-Reply-To: <1619434136-17760-1-git-send-email-lance.lmwang@gmail.com> References: <1619434136-17760-1-git-send-email-lance.lmwang@gmail.com> Subject: [FFmpeg-devel] [PATCH 4/6] tools/python: add help script to get sess_config X-BeenThere: ffmpeg-devel@ffmpeg.org X-Mailman-Version: 2.1.29 Precedence: list List-Id: FFmpeg development discussions and patches List-Unsubscribe: , List-Archive: List-Post: List-Help: List-Subscribe: , Reply-To: FFmpeg development discussions and patches Cc: Limin Wang MIME-Version: 1.0 Errors-To: ffmpeg-devel-bounces@ffmpeg.org Sender: "ffmpeg-devel" X-TUID: vDmOQMqPoN9r From: Limin Wang Please note the byte order of the hex data is in normal order. Signed-off-by: Limin Wang --- tools/python/tf_sess_config.py | 44 ++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 44 insertions(+) create mode 100644 tools/python/tf_sess_config.py diff --git a/tools/python/tf_sess_config.py b/tools/python/tf_sess_config.py new file mode 100644 index 0000000..e4e38bd --- /dev/null +++ b/tools/python/tf_sess_config.py @@ -0,0 +1,44 @@ +# Copyright (c) 2021 +# +# This file is part of FFmpeg. +# +# FFmpeg is free software; you can redistribute it and/or +# modify it under the terms of the GNU Lesser General Public +# License as published by the Free Software Foundation; either +# version 2.1 of the License, or (at your option) any later version. +# +# FFmpeg is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU +# Lesser General Public License for more details. +# +# You should have received a copy of the GNU Lesser General Public +# License along with FFmpeg; if not, write to the Free Software +# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA +# ============================================================================== + +# verified with Python 3.6.8 on CentOS 7.2 +import tensorflow as tf + +visible_device_list = '0' # use , separator for more GPUs like '0, 1' +per_process_gpu_memory_fraction = 0.9 # avoid out of memory +intra_op_parallelism_threads = 2 # default in tensorflow +inter_op_parallelism_threads = 5 # default in tensorflow + +gpu_options = tf.compat.v1.GPUOptions( + per_process_gpu_memory_fraction = per_process_gpu_memory_fraction, + visible_device_list = visible_device_list, + allow_growth = True) + +config = tf.compat.v1.ConfigProto( + allow_soft_placement = True, + log_device_placement = False, + intra_op_parallelism_threads = intra_op_parallelism_threads, + inter_op_parallelism_threads = inter_op_parallelism_threads, + gpu_options = gpu_options) + +s = config.SerializeToString() +# print(list(map(hex, s))) # print by json if need + +b = ''.join(format(b,'02x') for b in s) +print('0x%s' % b) # print by hex format