def defineNetwork( self ): print( 'define network' ) input_tensor = Input( shape = ( self.dim, self.dim, self.nFlowMaps * self.flowSteps ) ) model = BaseModel( input_tensor = input_tensor, weights = None, classes = self.classes ) optimizer = SGD( lr=1e-2, momentum = 0.9, decay=1e-4 ) model.compile( loss = 'categorical_crossentropy', optimizer = optimizer, metrics = [ 'acc' ] ) return model