move masks to cpu

[geraseva]

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.