from local

This commit is contained in:
dohe0342 2023-02-03 01:32:59 +09:00
parent 8fe989ee6b
commit bbcce34eeb
2 changed files with 7 additions and 0 deletions

View File

@ -54,6 +54,13 @@ def get_parser():
parser = argparse.ArgumentParser(
formatter_class=argparse.ArgumentDefaultsHelpFormatter
)
parser.add_argument(
"--wandb",
type=str2bool,
default=False,
help="Should various information be logged in tensorboard.",
)
parser.add_argument(
"--world-size",