Open juliangamble opened 1 year ago
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 PtNDArrayEx
.
@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 a specification - it seems to be this class here: https://github.com/apache/mxnet/blob/master/src/operator/contrib/multibox_prior.cc Please help me out if you know a better one.
In terms of measuring if it is working - I'm looking in here - and not finding anything that corresponds: https://github.com/apache/mxnet/tree/master/tests/cpp/operator
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 OptimizerTest
.
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. https://github.com/deepjavalibrary/djl/pull/2715 The two different unit tests nearly match up, but not quite - so I'm asking for some help 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