Skip to content

Commit 171114e

Browse files
committed
Adapt to latest changes in TF Java
1 parent d09e5a9 commit 171114e

File tree

2 files changed

+1
-3
lines changed

2 files changed

+1
-3
lines changed

tensorflow-examples/src/main/java/org/tensorflow/model/examples/mnist/CnnMnist.java

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -308,8 +308,6 @@ public static void main(String[] args) {
308308
logger.info("Trained model");
309309

310310
test(session, minibatchSize, dataset);
311-
312311
}
313-
314312
}
315313
}

tensorflow-examples/src/main/java/org/tensorflow/model/examples/mnist/SimpleMnist.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -83,7 +83,7 @@ public void run() {
8383
try (Session session = new Session(graph)) {
8484

8585
// Initialize variables
86-
session.runInit();
86+
session.run(tf.init());
8787

8888
// Train the model
8989
for (ImageBatch trainingBatch : dataset.trainingBatches(TRAINING_BATCH_SIZE)) {

0 commit comments

Comments
 (0)