from local
This commit is contained in:
parent
91b9cc29a7
commit
8afcd618c6
Binary file not shown.
@ -1344,6 +1344,9 @@ def main():
|
|||||||
logging.info("save arguments to config.yaml...")
|
logging.info("save arguments to config.yaml...")
|
||||||
save_args(args)
|
save_args(args)
|
||||||
|
|
||||||
|
if args.wandb: wb = wandb.init(project="d2v-T", entity="dohe0342", config=vars(args))
|
||||||
|
else: wb = None
|
||||||
|
|
||||||
world_size = args.world_size
|
world_size = args.world_size
|
||||||
assert world_size >= 1
|
assert world_size >= 1
|
||||||
if world_size > 1:
|
if world_size > 1:
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user