Commit 57087432 authored by Oleg Dzhimiev's avatar Oleg Dzhimiev

minor changes

parent 0452a446
...@@ -93,16 +93,16 @@ def lrelu(x): ...@@ -93,16 +93,16 @@ def lrelu(x):
def network(input): def network(input):
fc1 = slim.fully_connected(input,1024,activation_fn=lrelu,scope='g_fc1') fc1 = slim.fully_connected(input,1024,activation_fn=lrelu,scope='g_fc1')
#fc2 = slim.fully_connected(fc1, 2,activation_fn=lrelu,scope='g_fc2') fc2 = slim.fully_connected(fc1, 2,activation_fn=lrelu,scope='g_fc2')
#return fc2 return fc2
fc2 = slim.fully_connected(fc1, 1024,activation_fn=lrelu,scope='g_fc2') #fc2 = slim.fully_connected(fc1, 1024,activation_fn=lrelu,scope='g_fc2')
fc3 = slim.fully_connected(fc2, 512,activation_fn=lrelu,scope='g_fc3') #fc3 = slim.fully_connected(fc2, 512,activation_fn=lrelu,scope='g_fc3')
fc4 = slim.fully_connected(fc3, 8,activation_fn=lrelu,scope='g_fc4') #fc4 = slim.fully_connected(fc3, 8,activation_fn=lrelu,scope='g_fc4')
fc5 = slim.fully_connected(fc4, 4,activation_fn=lrelu,scope='g_fc5') #fc5 = slim.fully_connected(fc4, 4,activation_fn=lrelu,scope='g_fc5')
fc6 = slim.fully_connected(fc5, 2,activation_fn=lrelu,scope='g_fc6') #fc6 = slim.fully_connected(fc5, 2,activation_fn=lrelu,scope='g_fc6')
return fc6 #return fc6
sess = tf.Session() sess = tf.Session()
......
...@@ -69,7 +69,7 @@ def lrelu(x): ...@@ -69,7 +69,7 @@ def lrelu(x):
def network(input): def network(input):
fc1 = slim.fully_connected(input,512,activation_fn=lrelu,scope='g_fc1') fc1 = slim.fully_connected(input,1024,activation_fn=lrelu,scope='g_fc1')
fc2 = slim.fully_connected(fc1, 2,activation_fn=lrelu,scope='g_fc2') fc2 = slim.fully_connected(fc1, 2,activation_fn=lrelu,scope='g_fc2')
return fc2 return fc2
......
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