riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 33 Packages Projects Releases Wiki Activity

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

Browse Source 
when `NEPTUNE_RUN_ID` environmetnt variable is set, neptune will log into the previous run with id `NEPTUNE_RUN_ID`
This commit is contained in:
fcakyon 2021-08-30 16:59:17 +03:00 committed by GitHub
parent f4f4e6b2d3
commit 6f3c99acca
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/transformers/integrations.py
View File

@ -728,6 +728,7 @@ class NeptuneCallback(TrainerCallback):
api_token=os.getenv("NEPTUNE_API_TOKEN"),
mode=os.getenv("NEPTUNE_CONNECTION_MODE", "async"),
name=os.getenv("NEPTUNE_RUN_NAME", None),
run=os.getenv("NEPTUNE_RUN_ID", None),
)
combined_dict = args.to_dict()
if hasattr(model, "config") and model.config is not None: