-
Notifications
You must be signed in to change notification settings - Fork 642
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Is it possible to train a PyTorch SSD model on an M1 Mac - or is this not yet implemented? PtNDArrayEx.multiBoxPrior(PtNDArrayEx.java:697) UnsupportedOperationException: Not implemented #2693
Comments
MXNet has several helper operators specific to SSD and they were used as part of the DJL SSD model you are using. Unfortunately, MXNet doesn't support M1 and the model doesn't run on PyTorch. If you are interested in contributing here, you could build an implementation of SSD that does not rely on those operators or you could add the missing implementations as part of |
@zachgk thanks for getting back to me. Thanks for creating an opportunity to contribute. I'm sizing it up - and working out a specification and way to measure if it is working. In terms of measuring if it is working - I'm looking in here - and not finding anything that corresponds: Can you help me out with how you would measure a working implementation? |
Probably the easiest way to test whether it is working is to use a hard-coded value for inputs and outputs. We have some examples in So, find a known sample data and then you can put it into the integration suite so it is run in all engines. This way, it ensures that all engines have matching behavior (including between the MXNet version and your new implementation). It also ensures that the behavior won't change because it would require also changing the values in the test |
I'll get back to you - I'm writing a test. |
I've done a pull request on this. |
Description
When running
TrainPikachuTest
on an M1 Mac I get the errorUnsupportedOperationException: Not implemented
Expected Behavior
The
TrainPikachuTest
runs as expected and a model is produced.Error Message
How to Reproduce?
Run the class
TrainPikachuTest
on an M1 MacSteps to reproduce
(Paste the commands you ran that produced the error.)
TrainPikachuTest
class withDJL_DEFAULT_ENGINE=PyTorch
What have you tried to solve it?
Environment Info
The text was updated successfully, but these errors were encountered: