"policy_norm" : {
      "@class" : "org.deeplearning4j.nn.conf.graph.LayerVertex",
      "layerConf" : {
        "cacheMode" : "NONE",
        "dataType" : "FLOAT",
        "epochCount" : 0,
        "iterationCount" : 0,
        "layer" : {
          "@class" : "org.deeplearning4j.nn.conf.layers.BatchNormalization",
          "activationFn" : {
            "@class" : "org.nd4j.linalg.activations.impl.ActivationSigmoid"
          },
          "beta" : 0.0,
          "biasInit" : 0.0,
          "biasUpdater" : null,
          "cnn2DFormat" : "NCHW",
          "constraints" : null,
          "cudnnAllowFallback" : true,
          "decay" : 0.9,
          "eps" : 1.0E-5,
          "gainInit" : 1.0,
          "gamma" : 1.0,
          "gradientNormalization" : "None",
          "gradientNormalizationThreshold" : 1.0,
          "idropout" : null,
          "iupdater" : {
            "@class" : "org.nd4j.linalg.learning.config.Sgd",
            "learningRate" : 0.001
          },
          "layerName" : "policy_norm",
          "lockGammaBeta" : false,
          "minibatch" : true,
          "nin" : 0,
          "nout" : 32,
          "regularization" : [ ],
          "regularizationBias" : [ ],
          "timeDistributedFormat" : null,
          "useLogStd" : true,
          "weightInitFn" : {
            "@class" : "org.deeplearning4j.nn.weights.WeightInitNormal"
          },
          "weightNoise" : null
        },
        "maxNumLineSearchIterations" : 5,
        "miniBatch" : true,
        "minimize" : true,
        "optimizationAlgo" : "STOCHASTIC_GRADIENT_DESCENT",
        "seed" : 1603508469355,
        "stepFunction" : null,
        "variables" : [ ]
      },
      "outputVertex" : false,
      "preProcessor" : null
    },