search

bryanyzhu / two-stream-pytorch

PyTorch implementation of two-stream networks for video action recognition
MIT License
568 stars 150 forks source link

The number of GPUs? #43

Closed WangShuu closed 2 years ago

WangShuu commented 4 years ago

Hello,I double how can i train this code with several GPUs? Since i just tried to run the code "main_sigle_gpu.py",only one GPU began to work.How to solve this problem?

bryanyzhu commented 4 years ago

Hi, if you want to train it with multiple GPUs, you can wrap the model with dataparallel like this line of code does,

https://github.com/bryanyzhu/two-stream-pytorch/blob/master/main.py#L179