fix normalization by number of examples

logix-project / logix

AI Logging for Interpretability and Explainability🔬

Apache License 2.0

74 stars 6 forks source link

fix normalization by number of examples #80

Closed hwijeen closed 8 months ago

hwijeen commented 8 months ago

This PR corrects how to get the number of examples from a batch. The bug is that len(data) does not give the number of examples as it is reshaped to 2d matrix of (bs*seq_len, feat_dim) in line 32.