From 3093dec599f04657a7ae6c7a9c97eba5c910259a Mon Sep 17 00:00:00 2001 From: Clea Dronne <66008479+clead6@users.noreply.github.com> Date: Thu, 1 Feb 2024 02:51:37 +0000 Subject: [PATCH] fix bug varautoencoder_mednist.ipynb (#1626) Bug in VAE tutorial, indexing of the data dictionaries train_datadict[:10] and test_datadict[:10] had been left there. ### Description In a previous commit, someone left indexes on the data dictionaries train_datadict[:10] and test_datadict[:10] to debug. I removed them so the code properly runs on the whole dataset. ### Checks Signed-off-by: Clea Dronne <66008479+clead6@users.noreply.github.com> --- modules/varautoencoder_mednist.ipynb | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/modules/varautoencoder_mednist.ipynb b/modules/varautoencoder_mednist.ipynb index 3645b01331..8e47e90fad 100644 --- a/modules/varautoencoder_mednist.ipynb +++ b/modules/varautoencoder_mednist.ipynb @@ -358,9 +358,9 @@ " ]\n", ")\n", "\n", - "train_ds = CacheDataset(train_datadict[:10], transforms, num_workers=num_workers)\n", + "train_ds = CacheDataset(train_datadict, transforms, num_workers=num_workers)\n", "train_loader = DataLoader(train_ds, batch_size=batch_size, shuffle=True, num_workers=num_workers)\n", - "test_ds = CacheDataset(test_datadict[:10], transforms, num_workers=num_workers)\n", + "test_ds = CacheDataset(test_datadict, transforms, num_workers=num_workers)\n", "test_loader = DataLoader(test_ds, batch_size=batch_size, shuffle=True, num_workers=num_workers)" ] },