Vertical-Federated-Learning icon indicating copy to clipboard operation
Vertical-Federated-Learning copied to clipboard

error occur

Open nanara1119 opened this issue 4 years ago • 0 comments

Hi

to try running which to see how it works but can not run source code Could you help me?

error occur below client.py -> def load_batch() -> x, y = list(self.data_loader)[point]

Original Traceback (most recent call last): File "C:\Dev\Python37\lib\site-packages\torch\utils\data_utils\worker.py", line 287, in _worker_loop data = fetcher.fetch(index) File "C:\Dev\Python37\lib\site-packages\torch\utils\data_utils\fetch.py", line 44, in fetch data = [self.dataset[idx] for idx in possibly_batched_index] File "C:\Dev\Python37\lib\site-packages\torch\utils\data_utils\fetch.py", line 44, in data = [self.dataset[idx] for idx in possibly_batched_index] File "C:\Users\LGH\AppData\Roaming\Python\Python37\site-packages\torchvision\datasets\mnist.py", line 131, in getitem img = Image.fromarray(img.numpy(), mode='L') File "C:\Dev\Python37\lib\site-packages\PIL\Image.py", line 2663, in fromarray size = shape[1], shape[0] IndexError: tuple index out of range

torch==1.9.0 torchaudio==0.9.0 torchvision==0.10.0

nanara1119 avatar Feb 15 '22 06:02 nanara1119