Skip to content

Commit 7f351c5

Browse files
author
Ye Shaokai
committed
avoid duplicate evaluation for po
1 parent aa3a913 commit 7f351c5

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

llava/train/train_dpo_new.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -222,6 +222,8 @@ class EK100EvalArguments:
222222
action_representation: str = "GT_random_narration_cut"
223223
n_narrations: int = -1
224224
test_type: str = 'base'
225+
learn_neighbor_actions: bool = False
226+
225227

226228
def maybe_zero_3(param, ignore_status=False, name=None):
227229
from deepspeed import zero
@@ -2000,8 +2002,8 @@ def make_inputs_require_grad(module, input, output):
20002002

20012003
# # this needs to be tested. But let's always evaluate the model after training.
20022004
# # we should also save the predictions into the experiment folder so we can analyze afterwards
2003-
if training_args.eval_steps!= trainer.state.global_step:
2004-
trainer.evaluate(eval_result_folder = training_args.output_dir)
2005+
# if training_args.eval_steps!= trainer.state.global_step:
2006+
# trainer.evaluate(eval_result_folder = training_args.output_dir)
20052007

20062008

20072009
if __name__ == "__main__":

0 commit comments

Comments
 (0)