neural-pipeline icon indicating copy to clipboard operation
neural-pipeline copied to clipboard

Let user choose a device by himself

Open toodef opened this issue 5 years ago • 0 comments

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.

toodef avatar Jun 03 '19 15:06 toodef