Exemple #1
0
 def correctness_function(self, model_batch_output, ideal_batch_output):
     return Network.onehot_correctness_function(self, model_batch_output,
                                                ideal_batch_output)
Exemple #2
0
 def update_weights(self, batch_of_inputs, batch_of_ideal_outputs,
                    epoch_index, batch_index):
     return Network.default_update_weights(self, batch_of_inputs,
                                           batch_of_ideal_outputs,
                                           epoch_index, batch_index)
 def correctness_function(self, model_batch_output, ideal_batch_output):
     model_batch_output.to(self.hardware)
     ideal_batch_output.to(self.hardware)
     return Network.onehot_correctness_function(self, model_batch_output,
                                                ideal_batch_output)
 def test(self, loader, correctness_function=None):
     return Network.default_test(
         self, loader, loss_function=self.classifier_loss_function)
 def forward(self, input_data):
     return Network.default_forward(self, input_data)