def test_any_scheduler_integration(self): train_gen = some_data_generator(OptimizerCheckpointTest.batch_size) valid_gen = some_data_generator(OptimizerCheckpointTest.batch_size) lr_scheduler = ExponentialLR(gamma=0.01) checkpointer = LRSchedulerCheckpoint(lr_scheduler, self.checkpoint_filename, period=1) self.model.fit_generator(train_gen, valid_gen, epochs=OptimizerCheckpointTest.epochs, steps_per_epoch=5, callbacks=[checkpointer])
def test_exponential_lr_integration(self): exponential_lr = ExponentialLR(gamma=0.01) self._fit_with_callback_integration(exponential_lr)
def test_any_scheduler_checkpoints(self): lr_scheduler = ExponentialLR(gamma=0.01) checkpointer = LRSchedulerCheckpoint(lr_scheduler, self.checkpoint_filename, period=1) self._test_checkpointer(checkpointer, lr_scheduler)