diff --git a/network.py b/network.py index 1c9f6a5..5503b85 100644 --- a/network.py +++ b/network.py @@ -17,11 +17,12 @@ class Config(): class Network: # TODO: Actually compile tensorflow properly - os.environ["TF_CPP_MIN_LOG_LEVEL"]="2" + #os.environ["TF_CPP_MIN_LOG_LEVEL"]="2" def custom_tanh(self, x, name=None): a = tf.Variable(2.00, tf.float32) return tf.scalar_mul(a, tf.tanh(x, name)) + def __init__(self, session): self.session = session diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000..b5b8177 --- /dev/null +++ b/requirements.txt @@ -0,0 +1,15 @@ +absl-py==0.1.10 +astor==0.6.2 +bleach==1.5.0 +gast==0.2.0 +grpcio==1.10.0 +html5lib==0.9999999 +Markdown==2.6.11 +numpy==1.14.1 +pkg-resources==0.0.0 +protobuf==3.5.1 +six==1.11.0 +tensorboard==1.6.0 +tensorflow==1.6.0 +termcolor==1.1.0 +Werkzeug==0.14.1