Skip to content

Commit

Permalink
bugfix
Browse files Browse the repository at this point in the history
  • Loading branch information
RabotniKuma committed Feb 25, 2024
1 parent 864601a commit 3b997bc
Show file tree
Hide file tree
Showing 3 changed files with 6 additions and 5 deletions.
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,7 @@ Using this library, you can:
# Setup
## With pip
```bash
pip install git+https://github.com/analokmaus/[email protected].0 # Stable
pip install git+https://github.com/analokmaus/[email protected].2 # Stable
pip install git+https://github.com/analokmaus/kuma_utils.git@master # Latest
```
**IMPORTANT**
Expand Down
7 changes: 4 additions & 3 deletions kuma_utils/torch/callbacks/logger.py
Original file line number Diff line number Diff line change
Expand Up @@ -51,9 +51,10 @@ def __init__(self,
def init_wandb(self, serial: str = None): # This is called in Trainer._train()
if not WANDB:
raise ValueError('wandb is not installed.')
if serial is not None and 'name' not in self.wandb_params.keys(): # No override
self.wandb_params.update({'name': serial})
wandb.init(**self.wandb_params)
wandb_params = self.wandb_params.copy()
if serial is not None and 'name' not in wandb_params.keys(): # No override
wandb_params.update({'name': serial})
wandb.init(**wandb_params)

def init_tensorboard(self, serial): # This is called in Trainer._train()
if self.tensorboard_dir is None and self.path is not None:
Expand Down
2 changes: 1 addition & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
[tool.poetry]
name = "kuma_utils"
version = "0.4.1"
version = "0.4.2"
description = ""
authors = ["Hiroshi Yoshihara"]
readme = "README.md"
Expand Down

0 comments on commit 3b997bc

Please sign in to comment.