def test_multi_collection_match(out_dir):
pre_test_clean_up()
hook = SessionHook(
out_dir=out_dir,
include_regex=["loss:0"],
include_collections=["default", "trial"],
save_config=SaveConfig(save_interval=2),
)
hook.get_collection("trial").include("loss:0")
helper_test_multi_collection_match(out_dir, hook)
def test_save_config_modes(out_dir):
pre_test_clean_up()
hook = SessionHook(out_dir=out_dir, include_collections=["weights"])
hook.get_collection("weights").save_config = {
modes.TRAIN: SaveConfigMode(save_interval=2),
modes.EVAL: SaveConfigMode(save_interval=3),
}
helper_save_config_modes(out_dir, hook)