Note that the changes outlined here: Change to how TTA() works affect this notebook.
I found changing
probs = np.mean(np.exp(log_preds),0)
to
probs = np.exp(log_preds)
fixed this. It would be great if someone could verify this is correct.
1 Like