def testConstructor(self): with self.assertRaisesRegexp(ValueError, "output_shapes must be of length 1 or *"): snt.nets.ConvNet2DTranspose(output_channels=self.output_channels, output_shapes=[], kernel_shapes=self.kernel_shapes, strides=self.strides, paddings=self.paddings) with self.assertRaisesRegexp(ValueError, "output_shapes must be of length 1 or *"): snt.nets.ConvNet2DTranspose(output_channels=self.output_channels, output_shapes=[[1, 2], [1, 2]], kernel_shapes=self.kernel_shapes, strides=[], paddings=self.paddings) with self.assertRaisesRegexp(KeyError, "Invalid initializer keys.*"): snt.nets.ConvNet2DTranspose( output_channels=self.output_channels, output_shapes=self.output_shapes, kernel_shapes=self.kernel_shapes, strides=self.strides, paddings=self.paddings, initializers={ "not_w": tf.truncated_normal_initializer(stddev=1.0) }) net = snt.nets.ConvNet2DTranspose(output_channels=self.output_channels, output_shapes=self.output_shapes, kernel_shapes=self.kernel_shapes, strides=self.strides, paddings=self.paddings) self.assertEqual(net.output_shapes, tuple(self.output_shapes) * len(self.output_channels)) self.assertEqual(len(net.layers), len(self.output_channels)) for i, layer in enumerate(net.layers): self.assertEqual(layer.output_channels, self.output_channels[i]) self.assertEqual(layer.stride, (1, ) + fill_shape(self.strides[0], 2) + (1, )) self.assertEqual(layer.kernel_shape, fill_shape(self.kernel_shapes[0], 2)) self.assertEqual(layer.padding, self.paddings[0]) self.assertEqual(layer.output_channels, net.output_channels[i]) self.assertEqual(layer.stride, (1, ) + fill_shape(net.strides[i], 2) + (1, )) self.assertEqual(layer.kernel_shape, fill_shape(net.kernel_shapes[i], 2)) self.assertEqual(layer.padding, net.paddings[i]) with self.assertRaisesRegexp(TypeError, "output_shapes must be iterable"): snt.nets.ConvNet2DTranspose(output_channels=self.output_channels, output_shapes=False, kernel_shapes=self.kernel_shapes, strides=self.strides, paddings=self.paddings)
def testConstructor(self): with self.assertRaisesRegexp(ValueError, "output_shapes must be of length 1 or *"): snt.nets.ConvNet2DTranspose(output_channels=self.output_channels, output_shapes=[], kernel_shapes=self.kernel_shapes, strides=self.strides, paddings=self.paddings) with self.assertRaisesRegexp(ValueError, "output_shapes must be of length 1 or *"): snt.nets.ConvNet2DTranspose(output_channels=self.output_channels, output_shapes=[[1, 2], [1, 2]], kernel_shapes=self.kernel_shapes, strides=[], paddings=self.paddings) with self.assertRaisesRegexp(KeyError, "Invalid initializer keys.*"): snt.nets.ConvNet2DTranspose( output_channels=self.output_channels, output_shapes=self.output_shapes, kernel_shapes=self.kernel_shapes, strides=self.strides, paddings=self.paddings, initializers={"not_w": tf.truncated_normal_initializer(stddev=1.0)}) net = snt.nets.ConvNet2DTranspose(output_channels=self.output_channels, output_shapes=self.output_shapes, kernel_shapes=self.kernel_shapes, strides=self.strides, paddings=self.paddings) self.assertEqual(net.output_shapes, tuple(self.output_shapes) * len(self.output_channels)) self.assertEqual(len(net.layers), len(self.output_channels)) for i, layer in enumerate(net.layers): self.assertEqual(layer.output_channels, self.output_channels[i]) self.assertEqual(layer.stride, (1,) + fill_shape(self.strides[0], 2) + (1,)) self.assertEqual(layer.kernel_shape, fill_shape(self.kernel_shapes[0], 2)) self.assertEqual(layer.padding, self.paddings[0]) self.assertEqual(layer.output_channels, net.output_channels[i]) self.assertEqual(layer.stride, (1,) + fill_shape(net.strides[i], 2) + (1,)) self.assertEqual(layer.kernel_shape, fill_shape(net.kernel_shapes[i], 2)) self.assertEqual(layer.padding, net.paddings[i]) with self.assertRaisesRegexp(TypeError, "output_shapes must be iterable"): snt.nets.ConvNet2DTranspose(output_channels=self.output_channels, output_shapes=False, kernel_shapes=self.kernel_shapes, strides=self.strides, paddings=self.paddings)
def testConstructor(self): net = snt.nets.ConvNet2D(output_channels=self.output_channels, kernel_shapes=self.kernel_shapes, strides=self.strides, paddings=self.paddings) self.assertEqual(len(net.layers), len(self.output_channels)) for i, layer in enumerate(net.layers): self.assertEqual(layer.output_channels, self.output_channels[i]) self.assertEqual(layer.stride, (1,) + fill_shape(self.strides[0], 2) + (1,)) self.assertEqual(layer.kernel_shape, fill_shape(self.kernel_shapes[0], 2)) self.assertEqual(layer.padding, self.paddings[0]) self.assertEqual(layer.output_channels, net.output_channels[i]) self.assertEqual(layer.stride, (1,) + fill_shape(net.strides[i], 2) + (1,)) self.assertEqual(layer.kernel_shape, fill_shape(net.kernel_shapes[i], 2)) self.assertEqual(layer.padding, net.paddings[i])