code: def convert_weights_to_numpy(weights_dict):
          
          return dict([(k.replace("arg:", "").replace("aux:", ""), v.asnumpy())
                       for k, v in weights_dict.items()])
docs: Convert weights to numpy