diff --git a/README.md b/README.md index 4df6dd8f4a..6a4dcab54d 100644 --- a/README.md +++ b/README.md @@ -80,13 +80,13 @@ def main(hparams): exp.argparse(hparams) exp.save() + model_save_path = '{}/{}/{}'.format(hparams.model_save_path, exp.name, exp.version) + # build model model = ExampleModel(hparams) # callbacks 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(filepath=model_save_path, save_function=None, save_best_only=True, verbose=True, monitor='val_acc', mode='min') # configure trainer