/csp.pytorch

A pytorch reimplementation of liuwei16/CSP, their trained keras weights are loaded in pytorch.

Primary LanguagePythonMIT LicenseMIT

Watchers