7272af90d4f5916f61f8bc5660e5605e730bbd78,examples/securenn/network_a.py,,,#,177

Before Change

      "prediction-client", [logits, y], prediction_client.receive_output)

  with tfe.Session() as sess:
    print("Init")
    sess.run(tf.global_variables_initializer(), tag="init")

    print("Training")
    sess.run(cache_updater, tag="training")

After Change

    batch_input_shape = [batch_size, flat_dim]
    // compute prediction
    model = tfe.keras.Sequential()
    model.add(tfe.keras.layers.Dense(ModelTrainer.HIDDEN_N,
                                     batch_input_shape=batch_input_shape))
    model.add(tfe.keras.layers.Activation("relu"))
    model.add(tfe.keras.layers.Dense(ModelTrainer.HIDDEN_N))
    model.add(tfe.keras.layers.Activation("relu"))

Instances