Join GitHub today
GitHub is home to over 28 million developers working together to host and review code, manage projects, and build software together.Sign up
Fix names of extract-layers of resnet101 #8998
After running resnet50 and resnet101 in example/ssd, the size of .params file for these two models are:
It shows the number of parameters for resnet101 is smaller than resnet50, which seems very weird.
For resnet50, the numbers of layers for different stages are [3, 4, 6, 3], so the extract-layers for SSD are 3+4+6-1=12 and 3+4+6+3-1=15，as the code in symbol_factory.py: