Ошибка типа: объекты h5py не могут быть засечены - PullRequest
Новая
       48

Ошибка типа: объекты h5py не могут быть засечены

0 голосов
/ 14 марта 2020

Я пытаюсь запустить реализацию PyTorch кода, который должен работать на наборе данных SBD .

Учебные метки изначально доступны в. bin-файл, который затем преобразуется в файлы HDF5 (.h5).

При запуске алгоритма я получаю сообщение об ошибке: «TypeError: объекты h5py не могут быть засечены»

Я думаю, что ошибка происходит от torch.utils.data.DataLoader.

Есть идеи, если мне не хватает какой-либо концепции здесь? Я читал, что травление обычно не является предпочтительным, но на данный момент мой набор данных только в формате HDF5.

Для справки, трассировка стека ошибки выглядит следующим образом: 

  File "G:\My Drive\Debvrat - shared\Codes\CASENet PyTorch Implementations\SBD-lijiaman\main.py", line 130, in <module>
    main()

  File "G:\My Drive\Debvrat - shared\Codes\CASENet PyTorch Implementations\SBD-lijiaman\main.py", line 85, in main
    win_feats5, win_fusion, viz, global_step)

  File "G:\My Drive\Debvrat - shared\Codes\CASENet PyTorch Implementations\SBD-lijiaman\train_val\model_play.py", line 31, in train
    for i, (img, target) in enumerate(train_loader):

  File "C:\Anaconda3\lib\site-packages\torch\utils\data\dataloader.py", line 819, in __iter__
    return _DataLoaderIter(self)

  File "C:\Anaconda3\lib\site-packages\torch\utils\data\dataloader.py", line 560, in __init__
    w.start()

  File "C:\Anaconda3\lib\multiprocessing\process.py", line 112, in start
    self._popen = self._Popen(self)

  File "C:\Anaconda3\lib\multiprocessing\context.py", line 223, in _Popen
    return _default_context.get_context().Process._Popen(process_obj)

  File "C:\Anaconda3\lib\multiprocessing\context.py", line 322, in _Popen
    return Popen(process_obj)

  File "C:\Anaconda3\lib\multiprocessing\popen_spawn_win32.py", line 89, in __init__
    reduction.dump(process_obj, to_child)

  File "C:\Anaconda3\lib\multiprocessing\reduction.py", line 60, in dump
    ForkingPickler(file, protocol).dump(obj)

  File "C:\Anaconda3\lib\site-packages\h5py\_hl\base.py", line 308, in __getnewargs__
    raise TypeError("h5py objects cannot be pickled")

  TypeError: h5py objects cannot be pickled

Я используя Conda версии 4.8.2, Python 3.7.4, PyTorch 1.0.0 с Cuda 10.2.89

Спасибо,

...