(vars, device="cuda")
| 74 | |
| 75 | @make_recursive_func |
| 76 | def todevice(vars, device="cuda"): |
| 77 | if isinstance(vars, torch.Tensor): |
| 78 | return vars.to(device) |
| 79 | elif isinstance(vars, str): |
| 80 | return vars |
| 81 | elif isinstance(vars, bool): |
| 82 | return vars |
| 83 | elif isinstance(vars, float): |
| 84 | return vars |
| 85 | elif isinstance(vars, int): |
| 86 | return vars |
| 87 | else: |
| 88 | raise NotImplementedError("invalid input type {} for tensor2numpy".format(type(vars))) |