[GitHub] [incubator-mxnet] QueensGambit edited a comment on issue #15337: Current MXNet-Dev master breaks loading of certain models

[GitBox <gi...@apache.org>]

QueensGambit edited a comment on issue #15337: Current MXNet-Dev master breaks loading of certain models
URL: https://github.com/apache/incubator-mxnet/issues/15337#issuecomment-513372662
 
 
   @roywei Thank your for the reply.
   Sorry, for the inconvenience there was apparently a `/` missing in the relative path of the config-files which I released. I just updated the .zip-Release files and it should work again for MXNet 1.4.1.
   
   This is the code how the model is currently loaded:
   https://github.com/QueensGambit/CrazyAra/blob/master/DeepCrazyhouse/src/domain/agent/neural_net_api.py#L66
   
   ```python
           sym = mx.sym.load(self.symbol_path)
           # https://github.com/apache/incubator-mxnet/issues/6951
           save_dict = mx.nd.load(self.params_path)
           arg_params = {}
           aux_params = {}
           for key, val in save_dict.items():
               param_type, name = key.split(":", 1)
               if param_type == "arg":
                   arg_params[name] = val
               if param_type == "aux":
                   aux_params[name] = val
           # set the context on CPU, switch to GPU if there is one available
           if ctx == "cpu":
               self.ctx = mx.cpu()
           elif ctx == "gpu":
               self.ctx = mx.gpu()
           else:
               raise Exception("Unavailable ctx mode given %s. You must either select 'cpu' or 'gpu'" % ctx)
           # define batch_size times executor objects which are used for inference
           # one executor object is used for the currently requested batch batch length
           # the requested batch length is variable and at maximum the given batch_size
           self.executors = []
           for i in range(batch_size):
               executor = sym.simple_bind(
                   ctx=self.ctx,
                   # add a new length for each size starting with 1
                   data=(i + 1, NB_CHANNELS_FULL, BOARD_HEIGHT, BOARD_WIDTH),
                   grad_req="null",
                   force_rebind=True,
               )
               executor.copy_params_from(arg_params, aux_params)
               self.executors.append(executor)
   ```

----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
 
For queries about this service, please contact Infrastructure at:
users@infra.apache.org


With regards,
Apache Git Services