Update README.md

This commit is contained in:
William Falcon
2019-03-31 16:35:10 -04:00
committed by GitHub
parent 9ff6108af1
commit 72239b4419
+1 -6
View File
@@ -84,12 +84,7 @@ def main(hparams):
model = ExampleModel(hparams)
# callbacks
early_stop = EarlyStopping(
monitor='val_acc',
patience=3,
mode='min',
verbose=True,
)
early_stop = EarlyStopping(monitor='val_acc', patience=3, mode='min', verbose=True)
model_save_path = '{}/{}/{}'.format(hparams.model_save_path, exp.name, exp.version)
checkpoint = ModelCheckpoint(