Error If launch experiment

[Dmitry Bleklov]

If I run following experiment from this repository https://github.com/Bekovmi/neuro-1, I will receive following error:
File "/home/Bekovmi/.conda/envs/py37/lib/python3.7/site-packages/catalyst/dl/main.py", line 41, in main
COMMANDS[args.command].main(args, uargs)
File "/home/Bekovmi/.conda/envs/py37/lib/python3.7/site-packages/catalyst/dl/scripts/run.py", line 117, in main
distributed_run(args.distributed, main_worker, args, unknown_args)
File "/home/Bekovmi/.conda/envs/py37/lib/python3.7/site-packages/catalyst/utils/distributed.py", line 188, in distributed_run
worker_fn(*args, **kwargs)
File "/home/Bekovmi/.conda/envs/py37/lib/python3.7/site-packages/catalyst/dl/scripts/run.py", line 112, in main_worker
runner.run_experiment(experiment)
File "/home/Bekovmi/.conda/envs/py37/lib/python3.7/site-packages/catalyst/core/runner.py", line 374, in run_experiment
self._run_event("on_exception")
File "/home/Bekovmi/.conda/envs/py37/lib/python3.7/site-packages/catalyst/core/runner.py", line 242, in _run_event
getattr(callback, event)(self.state)
File "/home/Bekovmi/.conda/envs/py37/lib/python3.7/site-packages/catalyst/core/callbacks/exception.py", line 17, in on_exception
raise exception
File "/home/Bekovmi/.conda/envs/py37/lib/python3.7/site-packages/catalyst/core/runner.py", line 361, in run_experiment
self._run_stage(stage)
File "/home/Bekovmi/.conda/envs/py37/lib/python3.7/site-packages/catalyst/core/runner.py", line 342, in _run_stage
self._run_epoch(stage=stage, epoch=state.epoch)
File "/home/Bekovmi/.conda/envs/py37/lib/python3.7/site-packages/catalyst/core/runner.py", line 329, in _run_epoch
self._run_loader(loader)
File "/home/Bekovmi/.conda/envs/py37/lib/python3.7/site-packages/catalyst/core/runner.py", line 289, in _run_loader
for i, batch in enumerate(loader):
File "/home/Bekovmi/.conda/envs/py37/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 345, in next
data = self._next_data()
File "/home/Bekovmi/.conda/envs/py37/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 856, in _next_data
return self._process_data(data)
File "/home/Bekovmi/.conda/envs/py37/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 881, in _process_data
data.reraise()
File "/home/Bekovmi/.conda/envs/py37/lib/python3.7/site-packages/torch/_utils.py", line 394, in reraise
raise self.exc_type(msg)
KeyError: Caught KeyError in DataLoader worker process 0.
Original Traceback (most recent call last):
File "/home/Bekovmi/.conda/envs/py37/lib/python3.7/site-packages/pandas/core/indexes/base.py", line 2646, in get_loc
return self._engine.get_loc(key)
File "pandas/_libs/index.pyx", line 111, in pandas._libs.index.IndexEngine.get_loc
File "pandas/_libs/index.pyx", line 138, in pandas._libs.index.IndexEngine.get_loc
File "pandas/_libs/hashtable_class_helper.pxi", line 1618, in pandas._libs.hashtable.PyObjectHashTable.get_item
File "pandas/_libs/hashtable_class_helper.pxi", line 1626, in pandas._libs.hashtable.PyObjectHashTable.get_item
KeyError: 17
During handling of the above exception, another exception occurred:
Traceback (most recent call last):
File "/home/Bekovmi/.conda/envs/py37/lib/python3.7/site-packages/torch/utils/data/_utils/worker.py", line 178, in _worker_loop
data = fetcher.fetch(index)
File "/home/Bekovmi/.conda/envs/py37/lib/python3.7/site-packages/torch/utils/data/_utils/fetch.py", line 44, in fetch
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/home/Bekovmi/.conda/envs/py37/lib/python3.7/site-packages/torch/utils/data/_utils/fetch.py", line 44, in
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/home/Bekovmi/.conda/envs/py37/lib/python3.7/site-packages/catalyst/data/dataset.py", line 52, in getitem
item = self.data[index]
File "/home/Bekovmi/.conda/envs/py37/lib/python3.7/site-packages/pandas/core/frame.py", line 2800, in getitem
indexer = self.columns.get_loc(key)
File "/home/Bekovmi/.conda/envs/py37/lib/python3.7/site-packages/pandas/core/indexes/base.py", line 2648, in get_loc
return self._engine.get_loc(self._maybe_cast_indexer(key))
File "pandas/_libs/index.pyx", line 111, in pandas._libs.index.IndexEngine.get_loc
File "pandas/_libs/index.pyx", line 138, in pandas._libs.index.IndexEngine.get_loc
File "pandas/_libs/hashtable_class_helper.pxi", line 1618, in pandas._libs.hashtable.PyObjectHashTable.get_item
File "pandas/_libs/hashtable_class_helper.pxi", line 1626, in pandas._libs.hashtable.PyObjectHashTable.get_item
KeyError: 17