flopml
/

mamba

flpelerin commited on Aug 24, 2024

Commit

9f7507f

1 Parent(s): 48c784d

Update file wandb.py

Files changed (1) hide show

wandb.py CHANGED Viewed

@@ -15,6 +15,8 @@ class Wandb:
         if self.name == '':
             self.name = 'run-' + RandomCode()
         os.environ['WANDB_API_KEY'] = self.api_key
         wandb.init(self.project, self.entity, self.name)

         if self.name == '':
             self.name = 'run-' + RandomCode()
+        print(f"wandb: {wandb}")
         os.environ['WANDB_API_KEY'] = self.api_key
         wandb.init(self.project, self.entity, self.name)