- Notifications
You must be signed in to change notification settings - Fork2
PyTorch to CoreML: Writing custom layers with Metal shaders - torch.nn.functional.grid_sample operation
NotificationsYou must be signed in to change notification settings
dneprDroid/pytorch-coreml-custom-layer-example
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
PyTorch to CoreML: Writing custom layers with Metal
cd Convertpython3 -m pip install -r requirements.txtpython3 convert.py -o /path/to/output/resulting/mlmodel/fileOpenTorchCoreMLDemo.xcodeproj in Xcode, build and run the app.
The code of the grid sample Metal-shader ishere.
More information about grid sample (warper) algorithm you can findhere.
About
PyTorch to CoreML: Writing custom layers with Metal shaders - torch.nn.functional.grid_sample operation
Topics
Resources
Uh oh!
There was an error while loading.Please reload this page.
Stars
Watchers
Forks
Releases
No releases published
Packages0
No packages published
Uh oh!
There was an error while loading.Please reload this page.