import tensorflow as tf

        self.loss = self.loss1 + self.loss2
        self.optimizer = tf.train.AdamOptimizer(learning_rate=learning_rate).minimize(self.loss)


# In[3]:


tf.reset_default_graph()
sess = tf.InteractiveSession()

size_layers = 128
learning_rate = 1e-3
num_layers = 2

