if return -1 from a hook that loop stopps · e89da15f18 - pytorch-lightning - Gitea: Git with a cup of tea

wassname/pytorch-lightning

mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-11 12:31:23 +08:00

if return -1 from a hook that loop stopps

This commit is contained in:

William Falcon

2019-04-21 13:38:50 -04:00

parent 004f015ee0

commit e89da15f18

2 changed files with 3 additions and 7 deletions

									
										pytorch_lightning/root_module/hooks.py
									
		-2
	
												View File
												
				@@ -19,5 +19,3 @@ class ModelHooks(torch.nn.Module):

				    def on_post_performance_check(self):

				        pass

				    def should_stop_epoch(self, data_batch):

				        return False