Commit 57087432 authored by Oleg Dzhimiev's avatar Oleg Dzhimiev
Browse files

minor changes

parent 0452a446

test_nn_feed.py

+8 −8
Original line number Diff line number Diff line
@@ -93,16 +93,16 @@ def lrelu(x): 
def network(input):



  fc1  = slim.fully_connected(input,1024,activation_fn=lrelu,scope='g_fc1')

  #fc2  = slim.fully_connected(fc1,     2,activation_fn=lrelu,scope='g_fc2')

  #return fc2

  fc2  = slim.fully_connected(fc1,     2,activation_fn=lrelu,scope='g_fc2')

  return 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')

  fc4  = slim.fully_connected(fc3,     8,activation_fn=lrelu,scope='g_fc4')

  fc5  = slim.fully_connected(fc4,     4,activation_fn=lrelu,scope='g_fc5')

  fc6  = slim.fully_connected(fc5,     2,activation_fn=lrelu,scope='g_fc6')

  #fc2  = slim.fully_connected(fc1,  1024,activation_fn=lrelu,scope='g_fc2')

  #fc3  = slim.fully_connected(fc2,   512,activation_fn=lrelu,scope='g_fc3')

  #fc4  = slim.fully_connected(fc3,     8,activation_fn=lrelu,scope='g_fc4')

  #fc5  = slim.fully_connected(fc4,     4,activation_fn=lrelu,scope='g_fc5')

  #fc6  = slim.fully_connected(fc5,     2,activation_fn=lrelu,scope='g_fc6')



  return fc6

  #return fc6





sess = tf.Session()

test_nn_infer.py

+1 −1
Original line number Diff line number Diff line
@@ -69,7 +69,7 @@ def lrelu(x): 


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')

  return fc2