Update - I tried changing the criterion algorithm to SGD (no weights changed) and then Adam (weights change and it’s actually learning)
Is there an issue with either of these optimisations or am I missing something?