jwyang / fpn.pytorch

Pytorch implementation of Feature Pyramid Network (FPN) for Object Detection
MIT License
952 stars 221 forks source link

confused about the BN layer #7

Open iSmida opened 6 years ago

iSmida commented 6 years ago

Hi,thanks for your works. The BN layer in pytorch is
class torch.nn.BatchNorm2d(num_features, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True) in your code you choose the default parameters. Is that right for object detection with a small batchsize?