c1c388551ffc4b10de78d46e8065e7cdea4deb3b,tests/framework/model/test_model.py,ModelTest,test_correct_optim_calls_1_batch_per_step,#ModelTest#,174

Before Change

    def test_correct_optim_calls_1_batch_per_step(self):
        train_generator = some_data_tensor_generator(ModelTest.batch_size)

        _ = self.mocked_optim_model.fit_generator(train_generator,
                                                  None,
                                                  epochs=1,
                                                  steps_per_epoch=1,
                                                  batches_per_step=1)

        self.assertEqual(1, self.mocked_optimizer.step.call_count)
        self.assertEqual(1, self.mocked_optimizer.zero_grad.call_count)

After Change

                                   self.loss_function,
                                   batch_metrics=self.batch_metrics,
                                   epoch_metrics=self.epoch_metrics)
        mocked_optim_model.fit_generator(train_generator, None, epochs=1, steps_per_epoch=1, batches_per_step=1)

        self.assertEqual(1, mocked_optimizer.step.call_count)
        self.assertEqual(1, mocked_optimizer.zero_grad.call_count)

Instances