def test_trainer_interrupted_flag(tmpdir): """Test the flag denoting that a user interrupted training.""" model = DictHparamsModel({'in_features': 28 * 28, 'out_features': 10}) class InterruptCallback(Callback): def __init__(self): super().__init__() def on_batch_start(self, trainer, pl_module): raise KeyboardInterrupt interrupt_callback = InterruptCallback() trainer = Trainer( callbacks=[interrupt_callback], max_epochs=1, val_percent_check=0.1, train_percent_check=0.2, progress_bar_refresh_rate=0, logger=False, default_root_dir=tmpdir, ) assert not trainer.interrupted trainer.fit(model) assert trainer.interrupted
def test_trainer_interrupted_flag(tmpdir): """Test the flag denoting that a user interrupted training.""" model = DictHparamsModel({'in_features': 28 * 28, 'out_features': 10}) class InterruptCallback(Callback): def __init__(self): super().__init__() def on_batch_start(self, trainer, pl_module): raise KeyboardInterrupt interrupt_callback = InterruptCallback() trainer_options = { 'callbacks': [interrupt_callback], 'max_epochs': 1, 'val_percent_check': 0.1, 'train_percent_check': 0.2, 'progress_bar_refresh_rate': 0, 'logger': False, 'default_root_dir': tmpdir, } trainer = Trainer(**trainer_options) assert not trainer.interrupted trainer.fit(model) assert trainer.interrupted
def test_hparams_save_load(tmpdir): model = DictHparamsModel({'in_features': 28 * 28, 'out_features': 10, 'failed_key': lambda x: x}) trainer = Trainer( default_root_dir=tmpdir, max_epochs=1, ) # fit model result = trainer.fit(model) assert result == 1 # try to load the model now pretrained_model = tutils.load_model_from_checkpoint( trainer.checkpoint_callback.dirpath, module_class=DictHparamsModel )
def test_hparams_save_load(tmpdir): model = DictHparamsModel({'in_features': 28 * 28, 'out_features': 10}) # logger file to get meta trainer_options = dict( default_save_path=tmpdir, max_epochs=1, ) # fit model trainer = Trainer(**trainer_options) result = trainer.fit(model) assert result == 1 # try to load the model now pretrained_model = tutils.load_model_from_checkpoint( trainer.checkpoint_callback.dirpath, module_class=DictHparamsModel)