Skip to content
GitLab
切换分支/标签
  1. 28 11月, 2021 3 次提交
  3. 26 3月, 2021 1 次提交
  5. 21 3月, 2021 1 次提交
  7. 29 12月, 2020 1 次提交
  9. 17 11月, 2020 1 次提交
    • Yuxin Wu's avatar
      FrozenBatchNorm half support · 84a09834
      由 Yuxin Wu 创作于 
      Summary:
doesn't affect speed, but may save a bit more memory

Pull Request resolved: https://github.com/fairinternal/detectron2/pull/482

Test Plan:
f233140419

f233150179

RetinaNet (on fair cluster): 0.14s/step, memory 3427M, mAP is good.

Reviewed By: theschnitz

Differential Revision: D25003984

Pulled By: ppwwyyxx

fbshipit-source-id: c251b6da1225e20b338775aae3e5882163a4015b
      84a09834
  11. 05 11月, 2020 1 次提交
  13. 29 10月, 2020 1 次提交
  15. 08 7月, 2020 1 次提交
  17. 02 6月, 2020 1 次提交
    • Yuxin Wu's avatar
      share the TORCH_VERSION constant · efe7bfbe
      由 Yuxin Wu 创作于 
      Summary: it appears too many times and will appear more

Reviewed By: rbgirshick

Differential Revision: D21828691

fbshipit-source-id: 8537aca6e234150793f8aeedb7dca4b60f8a97d9
      efe7bfbe
  19. 16 4月, 2020 1 次提交
    • Yuxin Wu's avatar
      use official SyncBatchNorm · 02478d2e
      由 Yuxin Wu 创作于 
      Reviewed By: rbgirshick

Differential Revision: D21050713

fbshipit-source-id: 81c95876e20cbc79b061f8d16108579484d878f3
      02478d2e
  21. 04 4月, 2020 1 次提交
    • Yuxin Wu's avatar
      use configurable for box head · 41ab4386
      由 Yuxin Wu 创作于 
      Reviewed By: rbgirshick

Differential Revision: D20808474

fbshipit-source-id: 7790a22f20c7c119f99a7d4a29246e52959c7f5c
      41ab4386
  23. 24 3月, 2020 1 次提交
    • Yuxin Wu's avatar
      Fix backward of new NaiveSyncBatchNorm · 7c69c051
      由 Yuxin Wu 创作于 
      Summary: the gradients do not backprop to inputs - causing a deadlock.

Reviewed By: rbgirshick

Differential Revision: D20531868

fbshipit-source-id: fac23221decbe9e5740ac699b96ac08fabe4f832
      7c69c051
  25. 19 3月, 2020 2 次提交
    • Yuxin Wu's avatar
      add back the assertion about SyncBatchNorm empty batch size · 4d024258
      由 Yuxin Wu 创作于 
      Summary: removed by mistake

Reviewed By: rbgirshick

Differential Revision: D20528069

fbshipit-source-id: be14399597fca9c2c506394f2fb370e8f014d7b2
      4d024258
    • Yuxin Wu's avatar
      less naive SyncBN · 2612e4a9
      由 Yuxin Wu 创作于 
      Summary:
Support different #images per worker, and 0 batch size. 5~10% slower than old implementation.

If needed, it should be used by calling the module manually - currently it cannot be controlled by a config.

Reviewed By: rbgirshick

Differential Revision: D20345669

fbshipit-source-id: ad1bf24235025ec2e6ff95635331cff52b487c1b
      2612e4a9
  27. 17 12月, 2019 1 次提交
  29. 08 12月, 2019 1 次提交
    • Yuxin Wu's avatar
      handle eps correctly when converting BN to FrozenBN · 71989dfa
      由 Yuxin Wu 创作于 
      Summary:
FrozenBN will add eps to var, so the conversion should not add it.

This does not affect existing models since we always load the weights after converting the model.

Reviewed By: rbgirshick

Differential Revision: D18788410

fbshipit-source-id: 5a9c6e454fd6cf63508fd2ae72fb68447e39d973
      71989dfa
  31. 13 10月, 2019 1 次提交
  33. 11 10月, 2019 1 次提交