def run(self, *args):
if self.source is None:<EOL><INDENT>self.model.summary()<EOL><DEDENT>else:<EOL><INDENT>x_data, y_data = next(iter(self.source.train_loader()))<EOL>self.model.summary(input_size=x_data.shape[<NUM_LIT:1>:])<EOL><DEDENT>
Print model summary