lol
This commit is contained in:
Carlos Chavez 2022-11-10 10:13:14 -05:00 committed by GitHub
parent 1ea31eb71e
commit c4b1860282
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions

View File

@ -509,7 +509,7 @@ def main():
os.makedirs(args.output_path, exist_ok=True) os.makedirs(args.output_path, exist_ok=True)
run = wandb.init(project=args.project_id, name=args.run_name, config=vars(args), dir=args.output_path+'/wandb') run = wandb.init(project=args.project_id, name=args.run_name, config=vars(args), dir=args.output_path+'/wandb')
# Inform the user of host, and various versions -- useful for debugging isseus. # Inform the user of host, and various versions -- useful for debugging issues.
print("RUN_NAME:", args.run_name) print("RUN_NAME:", args.run_name)
print("HOST:", socket.gethostname()) print("HOST:", socket.gethostname())
print("CUDA:", torch.version.cuda) print("CUDA:", torch.version.cuda)