def test_specnorm(self):
layer_test(DynastesDepthwiseConv2D,
kwargs={
'kernel_size': (3, 3),
'kernel_normalizer': 'spectral'
},
input_shape=(4, 16, 16, 5))
def test_simple(self):
layer_test(DynastesConv2D,
kwargs={
'filters': 3,
'kernel_size': (3, 3),
'activation': 'swish'
},
input_shape=(4, 16, 16, 3))
def test_simple(self):
layer_test(DynastesConv1D,
kwargs={
'filters': 3,
'kernel_size': 3,
'activation': 'mish'
},
input_shape=(5, 32, 3))
def test_specnorm(self):
layer_test(DynastesDepthwiseConv1D,
kwargs={
'kernel_size': 3,
'kernel_normalizer': 'spectral',
'kernel_regularizer': 'orthogonal'
},
input_shape=(5, 32, 3))
def test_specnorm(self):
layer_test(DynastesConv2D,
kwargs={
'filters': 7,
'kernel_size': (3, 3),
'kernel_normalizer': 'spectral',
'kernel_regularizer': 'orthogonal'
},
input_shape=(4, 16, 16, 5))
def test_specnorm(self):
layer_test(DynastesConv1D,
kwargs={
'filters': 3,
'kernel_size': 3,
'kernel_normalizer': 'spectral',
'use_wscale': True,
'kernel_regularizer': 'orthogonal'
},
input_shape=(5, 32, 3))
def test_specnorm(self):
layer_test(DynastesDense,
kwargs={
'units': 12,
'kernel_normalizer': 'spectral',
'use_wscale': True,
'wnorm': True,
'kernel_regularizer': 'orthogonal'
},
input_shape=(5, 32, 3))
def test_simple(self):
layer_test(DynastesConv2DTranspose,
kwargs={
'filters': 3,
'kernel_size': (3, 3),
'strides': (2, 2),
'padding': 'same'
},
input_shape=(None, 16, 16, 3),
expected_output_shape=(None, 32, 32, 3))
def test_simple(self):
layer_test(SelfAttentionBlock1D,
kwargs={
'attention_dim': 8,
'kernel_size': 3,
'output_dim': 16,
'num_heads': 4,
'multiquery_attention': True
},
input_shape=(5, 32, 3))
def test_simple(self):
layer_test(Wave2STFTLayer,
kwargs={
'n_bins': 128,
'hop_length': 64,
'padding': 64,
'hq': True
},
input_shape=(5, 32, 2),
expected_output_dtype=tf.complex128)
def test_specnorm(self):
layer_test(DynastesConv2DTranspose,
kwargs={
'filters': 7,
'kernel_size': (3, 3),
'strides': (2, 2),
'padding': 'same',
'kernel_normalizer': 'spectral',
'kernel_regularizer': 'orthogonal'
},
input_shape=(None, 16, 16, 5),
expected_output_shape=(None, 32, 32, 7))
def test_simple(self):
layer_test(DynastesDepthwiseConv2D,
kwargs={'kernel_size': (3, 3)},
input_shape=(4, 16, 16, 3))
def test_simple(self):
layer_test(Upsampling2D,
kwargs={'strides': (2, 2)},
input_shape=(None, 16, 16, 3),
expected_output_shape=(None, 32, 32, 3))
def test_simple(self):
layer_test(DynastesDepthwiseConv1D,
kwargs={'kernel_size': 3},
input_shape=(5, 32, 3))
def test_simple(self):
layer_test(DynastesDense, kwargs={'units': 12}, input_shape=(5, 32, 3))
def test_simple(self):
layer_test(Upsampling1D,
kwargs={'strides': 2},
input_shape=(None, 32, 3),
expected_output_shape=(None, 64, 3))
def test_simple(self):
layer_test(STFT2MelspectrogramLayer,
kwargs={'n_mels': 128},
input_shape=(5, 32, 128, 2),
input_dtype=tf.complex128,
expected_output_dtype=K.floatx())
def test_simple(self):
layer_test(SpecAugmentLayer, kwargs={}, input_shape=(5, 32, 80, 2))