Serge Durand

### ✨ Short description of the feature [tl;dr] Currently device is automatically inferred from model parameters and images / labels are transferred on this device. It could be interesting to...