Merge pull request karpathy#20 from lantiga/wandb-optional-import

Make wandb import conditioned to wandb_log=True
mikelittman · Jan 8, 2023 · e7cd674 · e7cd674
2 parents e53b9d2 + 09f1f45
commit e7cd674
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/README.md b/README.md
@@ -19,7 +19,6 @@ Dependencies:
 - `pip install tiktoken` for OpenAI's fast BPE code <3
 - `pip install wandb` for optional logging <3
 - `pip install tqdm`
-- `pip install networkx`
 
 ## usage
 

diff --git a/train.py b/train.py
@@ -13,7 +13,6 @@
 import time
 import math
 
-import wandb
 import numpy as np
 import torch
 from torch.nn.parallel import DistributedDataParallel as DDP
@@ -180,6 +179,7 @@ def get_lr(iter):
 
 # logging
 if wandb_log and gpu_id == 0:
+    import wandb
     wandb.init(project=wandb_project, name=wandb_run_name)
     wandb.config = {
         "batch_size": batch_size,