forked from keras-team/keras-io
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathcallbacks_master.py
72 lines (71 loc) · 2.2 KB
/
callbacks_master.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
"""
deliberately skipped: BaseLogger, History
kept: ProgbarLogger
"""
CALLBACKS_MASTER = {
'path': 'callbacks/',
'title': 'Callbacks API', # TODO
'toc': True,
'children': [
{
'path': 'base_callback',
'title': 'Base Callback class',
'generate': ['tensorflow.keras.callbacks.Callback']
},
{
'path': 'model_checkpoint',
'title': 'ModelCheckpoint',
'generate': ['tensorflow.keras.callbacks.ModelCheckpoint']
},
{
'path': 'tensorboard',
'title': 'TensorBoard',
'generate': ['tensorflow.keras.callbacks.TensorBoard']
},
{
'path': 'early_stopping',
'title': 'EarlyStopping',
'generate': ['tensorflow.keras.callbacks.EarlyStopping']
},
{ # LEGACY
'path': 'learning_rate_scheduler',
'title': 'LearningRateScheduler',
'generate': ['tensorflow.keras.callbacks.LearningRateScheduler']
},
{
'path': 'reduce_lr_on_plateau',
'title': 'ReduceLROnPlateau',
'generate': ['tensorflow.keras.callbacks.ReduceLROnPlateau']
},
{
'path': 'remote_monitor',
'title': 'RemoteMonitor',
'generate': ['tensorflow.keras.callbacks.RemoteMonitor']
},
{
'path': 'lambda_callback',
'title': 'LambdaCallback',
'generate': ['tensorflow.keras.callbacks.LambdaCallback']
},
{
'path': 'terminate_on_nan',
'title': 'TerminateOnNaN',
'generate': ['tensorflow.keras.callbacks.TerminateOnNaN']
},
{
'path': 'csv_logger',
'title': 'CSVLogger',
'generate': ['tensorflow.keras.callbacks.CSVLogger']
},
{
'path': 'progbar_logger',
'title': 'ProgbarLogger',
'generate': ['tensorflow.keras.callbacks.ProgbarLogger']
},
{
'path': 'backup_and_restore',
'title': 'BackupAndRestore',
'generate': ['tensorflow.keras.callbacks.experimental.BackupAndRestore']
},
]
}