标签： python-2.7 multiprocessing gpu pytorch
当我尝试在单GPU上运行AttnGAN网络时，它运行正常，但在多个GPU上，它会抛出错误：
RuntimeError: Expected hidden size (2, 24, 128), got (1L, 48L, 128L)
我该如何解决这个问题？
我是pytorch和处理多个GPU的新手，所以我可能会遗漏一些简单的东西。
感谢任何帮助。谢谢！