Skip to content
Snippets Groups Projects
Commit f15763b2 authored by Frisinghelli Daniel's avatar Frisinghelli Daniel
Browse files

Decreased number of epochs to 20.

parent df4ba10c
No related branches found
No related tags found
No related merge requests found
...@@ -34,7 +34,7 @@ LOGGER = logging.getLogger(__name__) ...@@ -34,7 +34,7 @@ LOGGER = logging.getLogger(__name__)
# network training configuration # network training configuration
TRAIN_CONFIG = { TRAIN_CONFIG = {
'checkpoint_state': {}, 'checkpoint_state': {},
'epochs': 20, 'epochs': 25,
'save': True, 'save': True,
'save_loaders': False, 'save_loaders': False,
'early_stop': False, 'early_stop': False,
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment