Update file logger.py
Browse files
logger.py
CHANGED
|
@@ -16,15 +16,10 @@ class Wandb:
|
|
| 16 |
self.name = 'run-' + RandomCode()
|
| 17 |
|
| 18 |
os.environ['WANDB_API_KEY'] = self.api_key
|
| 19 |
-
|
| 20 |
-
|
| 21 |
-
|
| 22 |
-
def init(self):
|
| 23 |
-
wandb.init(project=self.project, entity=self.entity, name=self.name, reinit=True, config={})
|
| 24 |
|
| 25 |
def __call__(self, args):
|
| 26 |
wandb.log(args)
|
| 27 |
|
| 28 |
-
|
| 29 |
def __exit__(self):
|
| 30 |
wandb.finish()
|
|
|
|
| 16 |
self.name = 'run-' + RandomCode()
|
| 17 |
|
| 18 |
os.environ['WANDB_API_KEY'] = self.api_key
|
| 19 |
+
wandb.init(project=self.project, entity=self.entity, name=self.name, reinit=True)
|
|
|
|
|
|
|
|
|
|
|
|
|
| 20 |
|
| 21 |
def __call__(self, args):
|
| 22 |
wandb.log(args)
|
| 23 |
|
|
|
|
| 24 |
def __exit__(self):
|
| 25 |
wandb.finish()
|