search

okankop / MFF-pytorch

Motion Fused Frames implementation in PyTorch, codes and pretrained models.
Other
131 stars 33 forks source link

Training code hangs for Jester Dataset #4

Closed saidwivedi closed 6 years ago

saidwivedi commented 6 years ago

When I run the training from scratch code for Jester dataset with the following command -

CUDA_VISIBLE_DEVICES=1,2 python main.py jester RGB --batch-size 32 --workers=1

The code hangs at https://github.com/okankop/MFF-pytorch/blob/master/main.py#L169 Am I missing something ?

I am using Python 3.6 and Pytorch 0.3.1

okankop commented 6 years ago

I have just fixed the issue. Now, the code is compatible for also RGB and Flow modalities. Could you please reclone the code and try?

saidwivedi commented 6 years ago

@okankop Yes I pulled the latest changes and it works. Thanks a lot for your quick response.

xiaofeiwu commented 5 years ago

@okankop I still meet this issue, could you let me know why?

okankop commented 5 years ago

@xiaofeiwu Could you please check if the getitem(self, index) in dataset.py is functioning properly? Your code might be stucking at lines 118 or 122 in dataset.py.