facebookresearch / facebookresearch/SlowFast

Training SLOWFAST_16x8_R101_50_50

Open
#74 0 comments 0 reactions 0 assignees View on GitHub
enhancement
Dominant language
Python
Stars
7.4k
Forks
1.3k
PR merge metrics
No merged PRs in 30d

Description

Hi, I have a question about training SLOWFAST_16x8_R101_50_50 model. If I understand correctly, this model requires a total input of 80 frames, 64 for fast branch and 16 for slow branch. So training this model consumes lots of GPU memory. For me, I can only use batch_size=3 per GPU (I use V100 with 16GB memory). However, I see in the [yaml](https://github.com/facebookresearch/SlowFast/blob/master/configs/Kinetics/c2/SLOWFAST_16x8_R101_50_50.yaml) file that you still use batch_size=8. Can you tell me how you accomplish this? And please correct me if my understanding is wrong. Thank you.

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.