损失 nan

一开始损失很大。我什么都没做，只是按照 Readme 指导。我使用单个 1650 台式机运行它。批量大小设置为 16，累积 2 个批次的梯度。