st_gcn_twostream.py报错运算需要同一类型

Posted crazyman33

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了st_gcn_twostream.py报错运算需要同一类型相关的知识，希望对你有一定的参考价值。

将其中forward()函数里面改为

      if True:
            m = torch.cat((torch.cuda.FloatTensor(N, C, 1, V, M).zero_(),
                           x[:, :, 1:] - x[:, :, :-1]), 2)
        else:
            m = torch.cat((torch.FloatTensor(N, C, 1, V, M).zero_(),
                           x[:, :, 1:] - x[:, :, :-1]), 2)

主要是不开GPU时会出现变量类型不一致，只能手动调节这里，没有入口参数

以上是关于st_gcn_twostream.py报错运算需要同一类型的主要内容，如果未能解决你的问题，请参考以下文章