Skip to content

Instantly share code, notes, and snippets.

@AFAgarap

AFAgarap/autoencoder-pytorch.ipynb

Last active April 28, 2024 12:30
Show Gist options
  • Star 24 You must be signed in to star a gist
  • Fork 7 You must be signed in to fork a gist
  • Save AFAgarap/4f8a8d8edf352271fa06d85ba0361f26 to your computer and use it in GitHub Desktop.
Save AFAgarap/4f8a8d8edf352271fa06d85ba0361f26 to your computer and use it in GitHub Desktop.
Download ZIP
PyTorch implementation of an autoencoder.
Display the source blob
Display the rendered blob
Raw
autoencoder-pytorch.ipynb
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
@Btwestyo
Copy link

I tried running this and I got this error:

raceback (most recent call last):
File "/Users/brianweston/Documents/Stanford_CS/CS231N_CNN/Project/cs231n_final_project/scripts/training/train_FMNIST.py", line 109, in
for batch_features, _ in train_loader:
File "/Users/brianweston/miniforge3/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 530, in next
data = self._next_data()
File "/Users/brianweston/miniforge3/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 570, in _next_data
data = self._dataset_fetcher.fetch(index) # may raise StopIteration
File "/Users/brianweston/miniforge3/lib/python3.8/site-packages/torch/utils/data/_utils/fetch.py", line 52, in fetch
return self.collate_fn(data)
File "/Users/brianweston/miniforge3/lib/python3.8/site-packages/torch/utils/data/_utils/collate.py", line 172, in default_collate
return [default_collate(samples) for samples in transposed] # Backwards compatibility.
File "/Users/brianweston/miniforge3/lib/python3.8/site-packages/torch/utils/data/_utils/collate.py", line 172, in
return [default_collate(samples) for samples in transposed] # Backwards compatibility.
File "/Users/brianweston/miniforge3/lib/python3.8/site-packages/torch/utils/data/_utils/collate.py", line 180, in default_collate
raise TypeError(default_collate_err_msg_format.format(elem_type))
TypeError: default_collate: batch must contain tensors, numpy arrays, numbers, dicts or lists; found <class 'PIL.Image.Image'>
(base) brianweston@MBP training % /Users/brianweston/miniforge3/bin/python /Users/brianweston/Documents/Stanford_CS/CS231N_CNN/Project/cs231n_final_project/scripts/training/train_FMNIST.py
Traceback (most recent call last):
File "/Users/brianweston/Documents/Stanford_CS/CS231N_CNN/Project/cs231n_final_project/scripts/training/train_FMNIST.py", line 109, in
for batch_features, _ in train_loader:
File "/Users/brianweston/miniforge3/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 530, in next
data = self._next_data()
File "/Users/brianweston/miniforge3/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 570, in _next_data
data = self._dataset_fetcher.fetch(index) # may raise StopIteration
File "/Users/brianweston/miniforge3/lib/python3.8/site-packages/torch/utils/data/_utils/fetch.py", line 52, in fetch
return self.collate_fn(data)
File "/Users/brianweston/miniforge3/lib/python3.8/site-packages/torch/utils/data/_utils/collate.py", line 172, in default_collate
return [default_collate(samples) for samples in transposed] # Backwards compatibility.
File "/Users/brianweston/miniforge3/lib/python3.8/site-packages/torch/utils/data/_utils/collate.py", line 172, in
return [default_collate(samples) for samples in transposed] # Backwards compatibility.
File "/Users/brianweston/miniforge3/lib/python3.8/site-packages/torch/utils/data/_utils/collate.py", line 180, in default_collate
raise TypeError(default_collate_err_msg_format.format(elem_type))
TypeError: default_collate: batch must contain tensors, numpy arrays, numbers, dicts or lists; found <class 'PIL.Image.Image'>
(base) brianweston@MBP training % /Users/brianweston/miniforge3/bin/python /Users/brianweston/Documents/Stanford_CS/CS231N_CNN/Project/cs231n_final_project/scripts/training/train_FMNIST.py
Traceback (most recent call last):
File "/Users/brianweston/Documents/Stanford_CS/CS231N_CNN/Project/cs231n_final_project/scripts/training/train_FMNIST.py", line 60, in
ax.imshow(np.array(img), cmap='gist_gray')
File "/Users/brianweston/miniforge3/lib/python3.8/site-packages/matplotlib/_api/deprecation.py", line 456, in wrapper
return func(*args, **kwargs)
File "/Users/brianweston/miniforge3/lib/python3.8/site-packages/matplotlib/init.py", line 1412, in inner
return func(ax, *map(sanitize_sequence, args), **kwargs)
File "/Users/brianweston/miniforge3/lib/python3.8/site-packages/matplotlib/axes/_axes.py", line 5488, in imshow
im.set_data(X)
File "/Users/brianweston/miniforge3/lib/python3.8/site-packages/matplotlib/image.py", line 715, in set_data
raise TypeError("Invalid shape {} for image data"
TypeError: Invalid shape (1, 28, 28) for image data

Any idea what the fix is?

@domattioli
Copy link

Kernel dies when running the Visualize Results, no error code, no changes. Everything prior to this successfully ran.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment