thats the issue im having, as far as I can tell, the model doesn’t run out of memory, there are no errors being shown, since I have 12 possible outputs(one hot matrix of 12), my default accuracy should be ~8.33%, which is what I’m getting even after 100 epochs when I say the model wont train. but it trains properly on the CPU of all machines and on the GPU of other machines.