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
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.