[cli] support on-the-fly training by loading pt model as nn.Module

[cli] support on-the-fly training by loading pt model as nn.Module #4317