Apex: nan loss with O1 and O2

[dhecloud (Andrew)]

Thanks! I found this link and followed the same method