update to notebook

Experiments/experiments.py

@@ -51,6 +51,9 @@ _, dataloader_chairs = get_chairs_dataloader(batch_size=32)
 for batch_chairs, labels_chairs in dataloader_chairs:
     break
 
+if not os.path.exists('../data/batch_chairs.pt'):
+    torch.save(batch_chairs, '../data/batch_chairs.pt')
+
 
 path_to_model_folder_chairs = '../trained_models/rendered_chairs/'
 list_expe = ['VAE_bs_64', 'VAE_bs_256', 'beta_VAE_bs_64', 'beta_VAE_bs_256', 'VAE_bs_64_ls_10_lr_1e_3',