Let user choose a device by himself

[toodef]

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.