from local

This commit is contained in:
dohe0342 2022-12-10 13:36:33 +09:00
parent c59c0052d0
commit f1dd8bc2eb
2 changed files with 2 additions and 2 deletions

View File

@ -106,7 +106,7 @@ def set_batch_count(model: Union[nn.Module, DDP], batch_count: float) -> None:
def add_rep_arguments(parser: argparse.ArgumentParser):
parser.add_argument(
"--wandb",
type=bool,
type=str2bool,
default=False,
help="Use wandb for MLOps",
)
@ -119,7 +119,7 @@ def add_rep_arguments(parser: argparse.ArgumentParser):
parser.add_argument(
"--multi-optim",
type=bool,
type=str2bool,
default=False,
help="use sperate optimizer (enc / dec)",
)