RuntimeError: “LayerNormKernelImpl“ not implemented for ‘Long‘

将GitHUB上下载的模块用于自己的模型,一定要注意输入数据的数据类型(dtype)是否满足模块的要求

第一步:数据类型为张量:train_inputs=torch.tensor(train_inputs);

第二步:数据中的元素类型:(根据自身需要:int、flaot16、float32)

train_inputs=torch.tensor(train_inputs,dtype=torch.float32)

你可能感兴趣的:(pytorch,深度学习,机器学习)