diff --git a/train.py b/train.py
index 83b82d3..6219033 100644
--- a/train.py
+++ b/train.py
@@ -15,7 +15,7 @@ from losses import d_logistic, d_logistic_r1_reg, g_logistic_non_saturating, g_l
 
 
 os.environ["CUDA_VISIBLE_DEVICES"] = "0"
-#wandb.init(config=tf.compat.v1.flags.FLAGS, sync_tensorboard=True, project='stylegan', name='test_2')
+wandb.init(config=tf.compat.v1.flags.FLAGS, sync_tensorboard=True, project='stylegan', name='test_conditioned_1')
 
 
 def initiate_models(g_params, d_params, use_custom_cuda):