Oleg Poyaganov has implemented a bridge from PyTorch to Core ML using coremltools. The Python package listing can be found here.
Only the nn module from Torch is currently supported but this still converts a range of different layers.
- Sequential
- ConcatTable
- SpatialConvolution
- ELU
- ReLU
- SpatialBatchNormalization
- Identity
- CAddTable
- SpatialFullConvolution
- SpatialSoftMax
- SpatialMaxPooling
- SpatialAveragePooling
- View
- Linear
- Tanh
- MulConstant
- SpatialZeroPadding
- SpatialReflectionPadding
- Narrow