MingLi
commited on
Commit
·
0ca782e
1
Parent(s):
0491592
adjust parameter
Browse files
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="
|
131 |
filename="squad-v2-{epoch:02d}-{train_loss:.2f}-{val_loss:.2f}",
|
132 |
save_last=True,
|
133 |
-
every_n_train_steps=
|
134 |
)
|
135 |
|
136 |
early_stop_callback = EarlyStopping(
|
137 |
-
monitor="
|
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=
|
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)
|