mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-09-09 11:32:07 +08:00
Add useful errors when model is not configured correctly (#1199)
* add check_model_configuration method * trying to fix errors * trying to fix tests * added test_epoch_end to lightning template * fix tests * fix new test after rebase * fix spelling * added more checks * updated formating * added tests * fixed CHANGELOG * Apply suggestions from code review * move test to new module * change check on configure_optimizers Co-authored-by: Nicki Skafte <nugginea@gmail.com> Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
This commit is contained in:
co-authored by
Nicki Skafte
Jirka Borovec
parent
ddb59130f8
commit
2912239fe6
@@ -232,6 +232,37 @@ class LightningTemplateModel(LightningModule):
|
||||
log.info('Test data loader called.')
|
||||
return self.__dataloader(train=False)
|
||||
|
||||
def test_step(self, batch, batch_idx):
|
||||
"""
|
||||
Lightning calls this during testing, similar to val_step
|
||||
:param batch:
|
||||
:return:val
|
||||
"""
|
||||
output = self.validation_step(batch, batch_idx)
|
||||
# Rename output keys
|
||||
output['test_loss'] = output.pop('val_loss')
|
||||
output['test_acc'] = output.pop('val_acc')
|
||||
|
||||
return output
|
||||
|
||||
def test_epoch_end(self, outputs):
|
||||
"""
|
||||
Called at the end of test to aggregate outputs, similar to validation_epoch_end
|
||||
:param outputs: list of individual outputs of each validation step
|
||||
:return:
|
||||
"""
|
||||
results = self.validation_step_end(outputs)
|
||||
|
||||
# rename some keys
|
||||
results['progress_bar'].update({
|
||||
'test_loss': results['progress_bar'].pop('val_loss'),
|
||||
'test_acc': results['progress_bar'].pop('val_acc'),
|
||||
})
|
||||
results['log'] = results['progress_bar']
|
||||
results['test_loss'] = results.pop('val_loss')
|
||||
|
||||
return results
|
||||
|
||||
@staticmethod
|
||||
def add_model_specific_args(parent_parser, root_dir): # pragma: no-cover
|
||||
"""
|
||||
|
||||
Reference in New Issue
Block a user