2martens/masterthesis
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Properly created global step

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-02-07 17:29:37 +01:00
parent 661e99c9f5
commit abf52a88b0
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/twomartens/masterthesis/aae/train.py
View File

@ -99,7 +99,7 @@ def train_mnist(folding_id: int, inlier_classes: Sequence[int], total_classes: i
y_real_z = k.ones(batch_size)
y_fake_z = k.zeros(batch_size)
sample = k.expand_dims(k.expand_dims(k.random_normal((64, zsize)), axis=1), axis=1)
global_step = k.variable(0)
global_step = tf.train.get_or_create_global_step()
encoder_loss_history = []
decoder_loss_history = []