MingLi commited on
Commit
0ca782e
·
1 Parent(s): 0491592

adjust parameter

Browse files
Files changed (1) hide show
  1. model.py +4 -4
model.py CHANGED
@@ -127,14 +127,14 @@ if __name__ == "__main__":
127
  checkpoint_callback = ModelCheckpoint(
128
  dirpath=args.checkpoint_dir,
129
  save_top_k=2,
130
- monitor="train_loss",
131
  filename="squad-v2-{epoch:02d}-{train_loss:.2f}-{val_loss:.2f}",
132
  save_last=True,
133
- every_n_train_steps=100,
134
  )
135
 
136
  early_stop_callback = EarlyStopping(
137
- monitor="val_loss",
138
  patience=3,
139
  check_finite=True,
140
  verbose=True,
@@ -151,7 +151,7 @@ if __name__ == "__main__":
151
  devices=1,
152
  logger=TensorBoardLogger("lightning_logs", name="squad-v2-bert"),
153
  enable_model_summary=True,
154
- val_check_interval=100,
155
  )
156
  if args.ckpt_path:
157
  trainer.fit(bert, data, ckpt_path=args.ckpt_path)
 
127
  checkpoint_callback = ModelCheckpoint(
128
  dirpath=args.checkpoint_dir,
129
  save_top_k=2,
130
+ monitor="val_loss",
131
  filename="squad-v2-{epoch:02d}-{train_loss:.2f}-{val_loss:.2f}",
132
  save_last=True,
133
+ every_n_train_steps=1000,
134
  )
135
 
136
  early_stop_callback = EarlyStopping(
137
+ monitor="train_loss",
138
  patience=3,
139
  check_finite=True,
140
  verbose=True,
 
151
  devices=1,
152
  logger=TensorBoardLogger("lightning_logs", name="squad-v2-bert"),
153
  enable_model_summary=True,
154
+ val_check_interval=0.25,
155
  )
156
  if args.ckpt_path:
157
  trainer.fit(bert, data, ckpt_path=args.ckpt_path)