finished lightning module

This commit is contained in:
William Falcon
2020-01-21 14:04:48 -05:00
parent 150a94543c
commit b351496001
@@ -95,6 +95,7 @@ class EarlyStopping(Callback):
early_stopping = EarlyStopping('val_loss')
Trainer(early_stop_callback=early_stopping)
"""
def __init__(self, monitor='val_loss',