def tune(self, input_batch_size=8, input_shape=32): """Tuning method.""" self.transform_output = naive_tune( self, (input_batch_size, self.in_channels, input_shape, input_shape), self.verbose, )
def tune(self, input_shape=4098): """Tuning method.""" self.transform_output = naive_tune(self, (input_shape, self.in_features), self.verbose)
def tune(self): """Tuning method.""" self.transform_output = naive_tune(self, (8, self.in_channels, 32, 32))
def tune(self): """Tuning method.""" self.transform_output = naive_tune(self, (4098, self.in_features))