f5a2b259b26bc173bf854cf5c8f306a1c21f15fe,txtgen/modules/networks_test.py,FeedForwardNetworkTest,test_feedforward,#FeedForwardNetworkTest#,19

Before Change

        _ = nn(tf.ones([64, 16, 16]))
        self.assertEqual(len(nn.trainable_variables),
                         len(hparams["layers"]) * 2)
        self.assertEqual(len(nn.layer_outputs), len(hparams["layers"]))

if __name__ == "__main__":
    tf.test.main()

After Change

        nn = FeedForwardNetwork(hparams=hparams)

        print(nn.hparams.todict())
        print(nn(tf.placeholder(dtype=tf.float32, shape=(None, 16))))
        //
        // self.assertEqual(len(nn.layers), len(hparams["layers"]))
        // _ = nn(tf.ones([64, 16, 16]))

Instances