thanks, i could run it. Can I have the TF/Keras equivalent as well?
Sample output that I’m seeing
Epoch 218 Loss: [ 20.28092194] Time: 0.12861180305480957
Epoch 219 Loss: [ 21.20872307] Time: 0.12335944175720215
Epoch 220 Loss: [ 20.74290848] Time: 0.11568808555603027
Epoch 221 Loss: [ 20.72460365] Time: 0.11843180656433105
Epoch 222 Loss: [ 19.11690903] Time: 0.11834907531738281
Epoch 223 Loss: [ 22.12939262] Time: 0.11621856689453125
Epoch 224 Loss: [ 19.81811905] Time: 0.11482071876525879
Epoch 225 Loss: [ 19.9168148] Time: 0.137786865234375