zoufeng2416 issues

Repositories
Issues
Comments

Results 2 issues of


                                            zoufeng2416

when I set the "--gpu-id", default=2, I meet this problem. only "--gpu-id", default=0. I can train the model.

RuntimeError: Expected tensor for 'out' to have the same device as tensor for argument #3 'batch2'; but device 2 does not equal 0 (while checking arguments for baddbmm) anyone could...

IndexError: The shape of the mask [845] at index 0 does not match the shape of the indexed tensor [5, 13, 13] at index 0

When I run the train_meta.py . I meet this error .if anyone can help me?