neural-pipeline
neural-pipeline copied to clipboard
Let user choose a device by himself
Now in np user do Trainer(..., device=torch.device('cuda)). It's good but bad when we try to pass unexpected data to device.
Need to make possible use callback for this + create some predefinitions.