改错笔记1

报错：

模型迭代计算传播导致结果变为NAN值

解决：

排查了好久发现

train_loader_pru = DataLoader(trainset_pru, batch_size=net_params["batch_size"], shuffle=True, drop_last=False,
                                    collate_fn=dataset_ori.train_collate)
    val_loader_pru = DataLoader(valset_pru, batch_size=net_params["batch_size"], shuffle=False, drop_last=False,
                                    collate_fn=dataset_ori.collate)
    test_loader_pru = DataLoader(testset_pru, batch_size=net_params["batch_size"], shuffle=False, drop_last=False,
                                     collate_fn=dataset_ori.collate)

将drop_last的值由True改为False就OK了，只需要丢弃最后一个批次，问题就解决了
Ps: 多小的一个问题啊，辛苦我排查那么久了