search

mpc001 / Lipreading_using_Temporal_Convolutional_Networks

ICASSP'22 Training Strategies for Improved Lip-Reading; ICASSP'21 Towards Practical Lipreading with Distilled and Efficient Models; ICASSP'20 Lipreading using Temporal Convolutional Networks
Other
395 stars 102 forks source link

How to run this model on 2 GPUs? #9

Closed CXiaoDing closed 3 years ago

CXiaoDing commented 4 years ago

Hi, thank you for your great work on lipreading. I want to implement this model on 2 GPUs, and I used nn.DataParallel to do it. But there have some problems in the function __averagebatch() in the model.py file, the error as follows: IndexError: index 16 is out of bounds for dimension 0 with size 16. Do you meet this problem? Thank you very much! @mpc001

mpc001 commented 4 years ago

Hi,

I didn't try multiple GPU. I suggest you could simply check forlengths, which seems out of bounds for tensor x in the function _average_batch.

zhanghousen commented 3 years ago

Hi, thank you for your great work on lipreading. I want to implement this model on 2 GPUs, and I used nn.DataParallel to do it. But there have some problems in the function __averagebatch() in the model.py file, the error as follows: IndexError: index 16 is out of bounds for dimension 0 with size 16. Do you meet this problem? Thank you very much! @mpc001

Have you solved this problem? I have the same problem @CXiaoDing