flpelerin commited on
Commit
8b32a10
·
1 Parent(s): dd73c60

Update file logger.py

Browse files
Files changed (1) hide show
  1. logger.py +1 -6
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
- self.init()
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()