Closed bupt229 closed 1 year ago
Fix the 'RuntimeError: indices should be either on cpu or on the same device as the indexed tensor (cpu)' problem for [GTN] [fastGTN] [MHNF] models.
Please feel free to remove inapplicable items for your PR.
Description
Fix the 'RuntimeError: indices should be either on cpu or on the same device as the indexed tensor (cpu)' problem for [GTN] [fastGTN] [MHNF] models.
Checklist
Please feel free to remove inapplicable items for your PR.
Changes