Skip to content

GitLab

Commit ad41c647 authored by Nimit Nigania's avatar Nimit Nigania
Browse files

change to add some logging

parent e38d570e
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
official/recommendation/ncf_keras_benchmark.py
View file @ ad41c647
...@@ -23,6 +23,10 @@ import time ...@@ -23,6 +23,10 @@ import time
from absl import flags from absl import flags
from absl.testing import flagsaver from absl.testing import flagsaver
os.environ["TF_CPP_VMODULE"] = "meta_optimizer=2"
os.environ["TF_CPP_MIN_VLOG_LEVEL"] = "1"
os.environ["TF_CPP_MIN_LOG_LEVEL"] = "0"
import tensorflow as tf # pylint: disable=g-bad-import-order import tensorflow as tf # pylint: disable=g-bad-import-order
from official.recommendation import ncf_common from official.recommendation import ncf_common
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment