在深度学习过程中遇到的一些bug及解决方法

PyTorch

ValueError: Unknown resampling filter (64).

将transform.Resize(64, 64)改成transform.Resize((64, 64))

Missing key(s) in state_dict:

在load_state_dict()中加上strict=False

DataLoader worker (pid(s) {}) exited unexpectedly

通常该异常会出现在内存不足的情况下,降低batch_size和线程数可以解决。或者在多线程任务中某一个线程exit()了也会抛出这样的异常。

RuntimeError: Expected tensor for argument #1 ‘indices’ to have scalar type Long; but got torch.cuda.FloatTensor instead (while checking arguments for embedding)

embedding输入必须是整型,要在输入数据后面加.long()

-------------本文结束感谢您的阅读-------------