Hello @seanmor5
I have the same problem, from chapter 7 when I try the cnn_trained_model_state the results are not the same as in the book :
09:03:50.990 [debug] Forwarding options: [compiler: EXLA] to JIT compiler
Epoch: 0, Batch: 150, accuracy: 0.5013453 loss: 7.5956130
Epoch: 1, Batch: 163, accuracy: 0.5018579 loss: 7.6527510
Epoch: 2, Batch: 176, accuracy: 0.5010152 loss: 7.6714020
Epoch: 3, Batch: 139, accuracy: 0.5034598 loss: 7.6697083
Epoch: 4, Batch: 152, accuracy: 0.5019404 loss: 7.6802869
And I have NaN in the model
[NaN],
[NaN],
[NaN],
[NaN],
[NaN],
[NaN],
[NaN],
[NaN],
...
]
>
}
}
```