Yep! you were correct … looked like my learning rate was too high - i originally set it to 1e-3 but then changed to 5e-5 and that seems to work:
{‘accuracy’: 0.341}
{‘accuracy’: 0.3574}
{‘accuracy’: 0.3592}
Thanks for the pointer Sylvain!