Use relative imports to fix issue with importing parallel trainer.