Skip to content

Commit

Permalink
Update TensorFlow script
Browse files Browse the repository at this point in the history
  • Loading branch information
lutzroeder committed Jul 3, 2024
1 parent f6f110c commit 56fd88e
Showing 1 changed file with 9 additions and 0 deletions.
9 changes: 9 additions & 0 deletions tools/tf_script.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,15 +4,24 @@
import logging
import os
import re
import sys
import google.protobuf # pylint: disable=import-error

logging.getLogger('tensorflow').setLevel(logging.ERROR)
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'

dup_stderr = os.dup(sys.stderr.fileno())
null = os.open(os.devnull, os.O_WRONLY)
os.dup2(null, sys.stderr.fileno())
os.close(null)

from tensorflow.core.framework import api_def_pb2 # pylint: disable=import-error,no-name-in-module,wrong-import-position
from tensorflow.core.framework import op_def_pb2 # pylint: disable=import-error,no-name-in-module,wrong-import-position
from tensorflow.core.framework import types_pb2 # pylint: disable=import-error,no-name-in-module,wrong-import-position

os.dup2(dup_stderr, sys.stderr.fileno())
os.close(dup_stderr)

def _read(path):
with open(path, 'r', encoding='utf-8') as file:
return file.read()
Expand Down

0 comments on commit 56fd88e

Please sign in to comment.