Hello, all!
I've created a feature request ticket ( None ) after I was surprised to find no previous mentions of this: Extending Task.init() behavior from training to inference (i.e. model.predict() ). See details on the ticket.
Looking forward to constructive feedback 😊