def load_model(self): if self.img_model == 'alexnet': img_output = img_alexnet_layers( self.img, self.batch_size, self.output_dim, self.stage, self.model_weights, val_batch_size=self.val_batch_size) else: raise Exception('cannot use such CNN model as ' + self.img_model) return img_output
def load_model(self): # print('Just tracking...') # after Ln42 if self.img_model == 'alexnet': # from args img_output = img_alexnet_layers(self.img, self.batch_size, self.output_dim, self.stage, self.model_weights, self.with_tanh, self.val_batch_size) else: raise Exception('cannot use such CNN model as ' + self.img_model) return img_output