Skip to content

Commit

Permalink
Disable tf logging
Browse files Browse the repository at this point in the history
  • Loading branch information
Rasmus Diederichsen committed Mar 6, 2018
1 parent 4883037 commit e7c911b
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion main.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,6 @@
.. moduleauthor:: Rasmus Diederichsen
'''

import sys
import numpy as np
# np.random.seed(1)
Expand All @@ -15,12 +14,14 @@
from os.path import join
import os.path
import os
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'

from model import VOModel
from utils import OptimizerSpec
from data_manager import DataManager
from performance_visualizer import PerformanceVisualizer


def make_parser():
'''Function returning parser is necessary for sphinx-argparse'''
tf_optimizers = {class_name[:-len('Optimizer')] for class_name in dir(tf.train) if 'Optimizer'
Expand Down Expand Up @@ -51,6 +52,7 @@ def make_parser():
help='Load checkpoint from checkpoints/deepvo.ckpt')
return parser


def main():

args = make_parser().parse_args()
Expand Down

0 comments on commit e7c911b

Please sign in to comment.