def test_vgg13(self):
        from dlpy.applications import VGG13
        model = VGG13(self.s)
        model.print_summary()

        # test random_crop and mutation
        model1 = VGG13(self.s, model_table='VGG16', n_classes=1000, n_channels=3,
                       width=224, height=224, scale=1,
                       offsets=None,
                       random_crop='unique',
                       random_flip='hv',
                       random_mutation='random')

        model1.print_summary()
        res1 = self.s.fetch(table=model1.model_name, sortby='_dllayerid_')
        print(res1)

        self.assertEqual(res1['Fetch'].iloc[10, 3], 4)
        self.assertEqual(res1['Fetch'].iloc[4, 3], 2)
Example #2
0
 def test_vgg13(self):
     from dlpy.applications import VGG13
     model = VGG13(self.s)
     model.print_summary()