DKJoey
Results
1
comments of
DKJoey
I met the same problem. Use mask = mask.cpu() will solve it.