spell0's picture
Fix BERNN probability and batch mapping
173a895