Change weights of a model wrapped in amp

3

I would not recommend to use apex/amp anymore, but to switch to the native implementation as described here.
One reason is the added flexibility for such use cases.