Update file wandb.py
Browse files
wandb.py
CHANGED
|
@@ -9,13 +9,15 @@ class Wandb:
|
|
| 9 |
def __init__(self, config: Config):
|
| 10 |
self.__dict__ = dict(config.__dict__)
|
| 11 |
|
|
|
|
|
|
|
|
|
|
| 12 |
if self.entity == '' or self.project == '' or self.api_key == '':
|
| 13 |
raise ValueError('Missing required values for either Entity, Project or API Key')
|
| 14 |
|
| 15 |
if self.name == '':
|
| 16 |
self.name = 'run-' + RandomCode()
|
| 17 |
|
| 18 |
-
print(f"wandb: {wandb}")
|
| 19 |
|
| 20 |
os.environ['WANDB_API_KEY'] = self.api_key
|
| 21 |
wandb.init(self.project, self.entity, self.name)
|
|
|
|
| 9 |
def __init__(self, config: Config):
|
| 10 |
self.__dict__ = dict(config.__dict__)
|
| 11 |
|
| 12 |
+
|
| 13 |
+
print(f"wandb: {wandb}")
|
| 14 |
+
|
| 15 |
if self.entity == '' or self.project == '' or self.api_key == '':
|
| 16 |
raise ValueError('Missing required values for either Entity, Project or API Key')
|
| 17 |
|
| 18 |
if self.name == '':
|
| 19 |
self.name = 'run-' + RandomCode()
|
| 20 |
|
|
|
|
| 21 |
|
| 22 |
os.environ['WANDB_API_KEY'] = self.api_key
|
| 23 |
wandb.init(self.project, self.entity, self.name)
|