def _build_model(self): self.model = model_builder("host", self.hetero_nn_param) self.model.set_transfer_variable(self.transfer_variable)
def _build_model(self): # return a hetero NN model with keras backend self.model = model_builder("guest", self.hetero_nn_param) self.model.set_transfer_variable(self.transfer_variable)