search

nmheim / torsk

An echo state network (ESN) for video prediction
10 stars 5 forks source link

`TorchImageDataset` broken #51

Open nmheim opened 5 years ago

nmheim commented 5 years ago 
Traceback (most recent call last):
  File "run.py", line 66, in <module>
    steps=1, step_length=2)
  File "/home/niklas/repos/toto/torsk/__init__.py", line 243, in train_predict_esn
    inputs, labels, pred_labels = dataset[idx]
  File "/home/niklas/repos/toto/torsk/data/torch_dataset.py", line 45, in __getitem__
    images = self._images[
AttributeError: 'TorchImageDataset' object has no attribute '_images'