small updates, dataset, low dim cvae

cvae.py

@@ -29,7 +29,7 @@ from PIL import Image
 from utils import str_to_bool
 from tf_utils import check_tf_version, allow_memory_growth, split_gpu_for_testing
 from load_models import load_generator, load_discriminator
-from losses import d_logistic, d_logistic_r1_reg, g_logistic_non_saturating, g_logistic_ns_pathreg, edge_loss, uv_mask_loss, bend_loss
+from losses import edge_loss, uv_mask_loss, bend_loss, kl_loss
 # from losses import collision_penalty as collision_penalty
 from losses import collision_loss as collision_penalty
 from custom_utils.custom_utils import batch_to_dict, load_model, plot_3d
@@ -39,8 +39,7 @@ from models.cvae_model import CVAE
 
 
 def init_wandb():
-    run = wandb.init(project='vae', name=config['name'],
-            notes=config['notes'])
+    run = wandb.init(project='vae', name=config['name'], notes=config['notes'])
     artifact = wandb.Artifact('code', type='code')
     artifact.add_file('cvae.py')
     artifact.add_file('models/cvae_model.py')
@@ -322,8 +321,6 @@ if __name__ == '__main__':
     parser.add_argument('--name', type=str)
     args = vars(parser.parse_args())
 
-    batch_size = config['batch_size']
-    
     smpl = SMPL()
 
     factory = DatasetFactory()