mirror/DeepCreamPy
1
0
Fork 0
mirror of https://github.com/Deepshift/DeepCreamPy.git synced 2024-11-28 11:09:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

rename network again

Browse Source
This commit is contained in:
deep pomf 2018-02-09 20:52:58 -05:00
parent 5cbafe7a5e
commit 1c89c5b153
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/train.py
View File

@ -21,7 +21,7 @@ def train():
local_completion = tf.placeholder(tf.float32, [BATCH_SIZE, LOCAL_SIZE, LOCAL_SIZE, 3]) local_completion = tf.placeholder(tf.float32, [BATCH_SIZE, LOCAL_SIZE, LOCAL_SIZE, 3])
is_training = tf.placeholder(tf.bool, []) is_training = tf.placeholder(tf.bool, [])
model = Network(x, mask, local_x, global_completion, local_completion, is_training, batch_size=BATCH_SIZE) model = Model(x, mask, local_x, global_completion, local_completion, is_training, batch_size=BATCH_SIZE)
sess = tf.Session() sess = tf.Session()
global_step = tf.Variable(0, name='global_step', trainable=False) global_step = tf.Variable(0, name='global_step', trainable=False)
epoch = tf.Variable(0, name='epoch', trainable=False) epoch = tf.Variable(0, name='epoch', trainable=False)