使用torch.nn.BatchNorm1d出现Tensor for argument #2 ‘weight‘ is on CPU, but expected it to on GPU错误
st_gcn网络定义中使用了torch.nn.BatchNorm1d，在运行网络时出现RuntimeError: Tensor for argument #2 'weight' is on CPU, but expected it to be on GPU (while checking arguments for cudnn_batch_norm)论坛中的解释是因为在forward函数中直接定义并使用了一个函数，但是源代码中是在__init__函数中定义，forward函数中再使用的githu
