Update README.md

This commit is contained in:
William Falcon authored and GitHub committed 2019-03-31 16:35:10 -04:00
1 parent 9ff6108af1
commit 72239b4419
1 file changed
+1 -6
+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(