pytorch
a6690bdb - fix input schema check for spatialbn

Commit
4 years ago
fix input schema check for spatialbn Summary: we were restricting it to 3, but in training we set up to 5, even that in practice we just need 3 since we don't recompute mean/var Test Plan: contrib tests for fakelowp Reviewed By: hl475 Differential Revision: D21905490 fbshipit-source-id: 48f61c7ba7d95f19d55d2f65514a517c1514ae88
Author
Hector Yuen
Parents
Loading