mirror of
https://github.com/webclinic017/drift.git
synced 2026-08-16 20:38:07 +00:00
feature: Added Weights and Biases configuration to the repo. (#48)
* feat: initial wandb configured. Sweep parameters aren't configured yet. * feat: Wandb logs now results. * feat: gitignore. * fix: Took out print() * feat: Changed default value of wandb to False. * feat: Added wandb to turn of automatically if there is no environment variable to start it (when we push it). Added environment configuration aswell. * fix(Dependencies): the package name seems to be python-dotenv Co-authored-by: Mark Aron Szulyovszky <mark.szulyovszky@gmail.com>
This commit is contained in:
co-authored by
Mark Aron Szulyovszky
parent
034bc1f213
commit
a7414eac23
@@ -0,0 +1,17 @@
|
||||
import wandb
|
||||
import os
|
||||
|
||||
|
||||
def get_wandb():
|
||||
from dotenv import load_dotenv
|
||||
load_dotenv()
|
||||
|
||||
''' 0. Login to Weights and Biases '''
|
||||
wsb_token = os.environ.get('WANDB_API_KEY')
|
||||
if wsb_token:
|
||||
wandb.login(key=wsb_token)
|
||||
return wandb
|
||||
else: return None #wandb.login()
|
||||
|
||||
# return wandb
|
||||
|
||||
Reference in New Issue
Block a user