facebookresearch / facebookresearch/SlowFast

Training on new dataset

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

Description

For training on my own dataset (which has 20 video classes), after the dataset has been prepared according to https://github.com/facebookresearch/SlowFast/blob/master/slowfast/datasets/DATASET.md, I need to change the number of classes in the yaml file? I am currently using SLOWFAST_8x8_R50.yaml configuration. Under the MODEL segment in YAML, I changed the number of classes to 20 from pre-existing 400.
Screen Shot 2020-04-04 at 12 09 32 AM
Is this the correct format to change the number of classes?
I printed the model format and got the following architecture for the ResnetBasicHead.
Screen Shot 2020-04-04 at 12 06 17 AM

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.