Commit d31b2ad6 authored by niclas.eich's avatar niclas.eich
Browse files

Necessary changes for tf20

parent bfec5291
......@@ -17,7 +17,7 @@ try:
CONDOR_ID = os.environ['CONDOR_ID']
except KeyError:
sys.exit('Error: Run this script with "pygpu %file"')
tf.logging.set_verbosity(tf.logging.ERROR)
tf.compat.v1.logging.set_verbosity(tf.compat.v1.logging.ERROR)
folder = 'train-%s' % CONDOR_ID # folder for training results
os.makedirs(folder)
......
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