Skip to content
This repository has been archived by the owner on Nov 3, 2023. It is now read-only.

Add WorldLogger to train_model script. #4369

Merged
merged 2 commits into from
Feb 25, 2022
Merged

Conversation

kauterry
Copy link
Contributor

Patch description

We'd like to dump the world logs for the "test" fold in the final evaluation during training. This way we can skip running the eval_model script while running training and evaluation back to back, and we can retrieve all the evaluation information from the world logs.

Testing steps

Added unittests in tests/test_train_model.py

Other information

)
with PathManager.open(task_log_report) as f:
json_lines = f.readlines()
assert len(json_lines) == 5
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

why is there only 5 lines instead of 10?

Copy link
Contributor

@jxmsML jxmsML left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM! a minor point.
Also it might be nice to take a look at the CI test failures.

@@ -647,8 +687,11 @@ def _run_eval(
reports = []

max_exs_per_worker = max_exs / (len(valid_worlds) * num_workers())
Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@jxmsML This is why we have only 5 lines instead of 10, cuz during multitask len(valid_worlds) > 1

@kauterry kauterry merged commit 7c2b199 into main Feb 25, 2022
@kauterry kauterry deleted the train_model_world_logging branch February 25, 2022 01:21
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants