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

yucornetto / MGMatting

This repository includes the official project of Mask Guided (MG) Matting, presented in our paper: Mask Guided Matting via Progressive Refinement Network

Other

332 stars 48 forks source link

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

Open priyatampintu opened 3 years ago

priyatampintu commented 3 years ago

Screenshot from 2021-09-20 14-25-08

yucornetto commented 3 years ago

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.