Update file logger.py

logger.py

@@ -16,9 +16,12 @@ class Wandb:
             self.name = 'run-' + RandomCode()
 
         os.environ['WANDB_API_KEY'] = self.api_key
-        wandb.init(self.project, self.entity, self.name, reinit=True, config={})
+        self.init()
 
 
+    def init(self):
+        
+        wandb.init(self.project, self.entity, self.name, reinit=True, config={})
 
     def __call__(self, args):
         wandb.log(args)