(varlist:list)
| 134 | |
| 135 | @torch.no_grad() |
| 136 | def tocuda(varlist:list): |
| 137 | out=[] |
| 138 | for var in varlist: |
| 139 | if isinstance(var,np.ndarray): |
| 140 | var=torch.from_numpy(var.copy()) |
| 141 | out.append(var.cuda()) |
| 142 | return out |
| 143 | |
| 144 | @torch.no_grad() |
| 145 | def tonumpy(varlist:list): |
no outgoing calls
no test coverage detected