facing error while training on single GPU. ('Generator' object has no attribute 'module')

Question

facing error while training on single GPU. ('Generator' object has no attribute 'module')

priyatampintu opened this issue 3 years ago · 1 comments

Answer 1 · 2021-09-29T00:31:03.000Z

It looks to me the problem is because when running the code on single GPU, there is some inconsistency about the dataparallel wrapping. Please try keep the dataparallel even when using single GPU.