mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-10-28 13:20:31 +08:00
Update README.md
This commit is contained in:
1 parent
9ff6108af1
commit
72239b4419
1 file changed
+1
-6
@@ -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(
|
||||
|
||||
Reference in new issue
Block a user