Unverified Commit 6f3c99ac authored by fcakyon's avatar fcakyon Committed by GitHub
Browse files

add ability to connect a neptune.ai run (#13319)

when `NEPTUNE_RUN_ID` environmetnt variable is set, neptune will log into the previous run with id `NEPTUNE_RUN_ID`
parent f4f4e6b2
...@@ -728,6 +728,7 @@ class NeptuneCallback(TrainerCallback): ...@@ -728,6 +728,7 @@ class NeptuneCallback(TrainerCallback):
api_token=os.getenv("NEPTUNE_API_TOKEN"), api_token=os.getenv("NEPTUNE_API_TOKEN"),
mode=os.getenv("NEPTUNE_CONNECTION_MODE", "async"), mode=os.getenv("NEPTUNE_CONNECTION_MODE", "async"),
name=os.getenv("NEPTUNE_RUN_NAME", None), name=os.getenv("NEPTUNE_RUN_NAME", None),
run=os.getenv("NEPTUNE_RUN_ID", None),
) )
combined_dict = args.to_dict() combined_dict = args.to_dict()
if hasattr(model, "config") and model.config is not None: if hasattr(model, "config") and model.config is not None:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment