From 10b905b445d42a9b75de6cbd133929eef9d5c26a Mon Sep 17 00:00:00 2001
From: z <51308183+nousr@users.noreply.github.com>
Date: Tue, 10 May 2022 09:52:50 -0700
Subject: [PATCH] smol typo (#81)

---
 train_diffusion_prior.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/train_diffusion_prior.py b/train_diffusion_prior.py
index c513c12..016b4e1 100644
--- a/train_diffusion_prior.py
+++ b/train_diffusion_prior.py
@@ -342,7 +342,7 @@ def main():
           RESUME,
           DPRIOR_PATH,
           config,
-          atgs.wandb_entity,
+          args.wandb_entity,
           args.wandb_project,
           args.learning_rate,
           args.max_grad_norm,