Skip to content

move masks to cpu#49

Open
geraseva wants to merge 5 commits intoarneschneuing:mainfrom
geraseva:main
Open

move masks to cpu#49
geraseva wants to merge 5 commits intoarneschneuing:mainfrom
geraseva:main

Conversation

@geraseva
Copy link

When running optimize.py and inpaint.py, I got the following error:

RuntimeError: indices should be either on cpu or on the same device as the indexed tensor (cpu)

I have corrected this by moving lig_mask tensor to cpu.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant