示例#1
0
 def test_serialization_with_layers(self):
   activation = activation_layers.LeakyReLU(alpha=0.1)
   layer = core.Dense(3, activation=activation)
   config = serialization.serialize(layer)
   # with custom objects
   deserialized_layer = serialization.deserialize(
       config, custom_objects={'LeakyReLU': activation})
   self.assertEqual(deserialized_layer.__class__.__name__,
                    layer.__class__.__name__)
   self.assertEqual(deserialized_layer.activation.__class__.__name__,
                    activation.__class__.__name__)
   # without custom objects
   deserialized_layer = serialization.deserialize(config)
   self.assertEqual(deserialized_layer.__class__.__name__,
                    layer.__class__.__name__)
   self.assertEqual(deserialized_layer.activation.__class__.__name__,
                    activation.__class__.__name__)
示例#2
0
    def test_invalid_usage(self):
        with self.assertRaises(ValueError):
            activations.get("unknown")

        # The following should be possible but should raise a warning:
        activations.get(activation_layers.LeakyReLU())