DNA Diffusion 跑模型debug

1.前向传播，应该是扩散过程没法跑，报的bug是“ Caught RuntimeError in pin memory thread for device 0.”。

我仔细看了报错的介绍发现是前面的dataloader部分有问题，而且多次出现pin_memory这个参数，于把pin_memory=True调整为pin_memory=False，之后成功运行。