8df5470cf8ce9f1a3b7e34fd6bd6b24be42523d8,src/data/dataset_merge.py,,main,#Any#,65

Before Change

  epoch = int(vec[1])
  print("loading",prefix, epoch)
  model = mx.mod.Module.load(prefix, epoch, context = ctx)
  model.bind(data_shapes=[("data", (args.batch_size, 3, image_size[0], image_size[1]))], label_shapes=[("softmax_label", (args.batch_size,))])
  rec_list = []
  for ds in include_datasets:
    path_imgrec = os.path.join(ds, "train.rec")

After Change

  //model.bind(data_shapes=[("data", (args.batch_size, 3, image_size[0], image_size[1]))], label_shapes=[("softmax_label", (args.batch_size,))])
  model = mx.mod.Module(symbol=sym, context=ctx, label_names = None)
  model.bind(data_shapes=[("data", (args.batch_size, 3, image_size[0], image_size[1]))])
  model.set_params(arg_params, aux_params)
  rec_list = []
  for ds in include_datasets:
    path_imgrec = os.path.join(ds, "train.rec")
    path_imgidx = os.path.join(ds, "train.idx")

Instances