Hi Stanislas, Did you consider nd/symbol.contrib.cond for conditional statement? https://mxnet.incubator.apache.org/versions/master/tutorials/control_flow/ControlFlowTutorial.html
Best, Haibin On Fri, Mar 8, 2019 at 10:29 Lauly, Stanislas <laul...@amazon.com.invalid> wrote: > Hi, > > About MXNET module, I need to create a model that use a different input > matrix depending on the actual input, is there a kind of “if” that can be > part of the computational graph? If not, I can always create multiple > models that share some params between them, but doing so does not sound > trivial. I found this: > > > https://discuss.mxnet.io/t/sharing-parameters-between-two-modules-through-arg-dict/1046 > > But it looks like it is not very efficient. Is there a better way to share > parameters between models? > > > Stanislas Lauly > Applied scientist – Amazon AI >