EnergyNewsKeyword/utils/info.py

15 lines
424 B
Python
Raw Permalink Normal View History

2023-04-19 13:16:27 +08:00
import logging
def log_versions():
import keras
logging.info("Keras version %s" % keras.__version__)
import numpy as np
logging.info("Numpy version %s" % np.__version__)
if keras.backend.backend() == 'theano':
import theano
logging.info("Theano version %s" % theano.__version__)
else:
import tensorflow
logging.info("Tensorflow version %s" % tensorflow.__version__)