Move sample to device.
(sample, device)
| 241 | |
| 242 | |
| 243 | def inplace_to_device(sample, device): |
| 244 | """Move sample to device.""" |
| 245 | sample["features"] = {k: v.to(device) for k, v in sample["features"].items()} |
| 246 | sample["indexer"] = {k: v.to(device) for k, v in sample["indexer"].items()} |
| 247 | if "labels" in sample.keys(): |
| 248 | sample["labels"] = sample["labels"].to(device) |
| 249 | return sample |
| 250 | |
| 251 | |
| 252 | def inplace_to_torch(sample): |
no outgoing calls
no test coverage detected