Allow for training "multi-port" surrogate

[m-bossart]

• Allow for full training process of multi-port surrogate.
• Make sure training efficiency (time) is not degraded.