Docs pre-release

Improved documentation and refactored train API (pytorch).

Awaiting results from several segmentation experiments ahead of public release (inc pytorch version).