In diversify/datautil/util.py, mydataset.__getitem__() yields (x, ctarget, dtarget, pctarget, pdtarget, index). However, in diversify/alg/algs/diversify.py, Diversify.update_d() unpackes the data with
all_d1 = minibatch[1].cuda().long()
all_c1 = minibatch[4].cuda().long()
I think it should be
all_d1 = minibatch[4].cuda().long()
all_c1 = minibatch[1].cuda().long()
to make it consistent.
In diversify/datautil/util.py,
mydataset.__getitem__()yields(x, ctarget, dtarget, pctarget, pdtarget, index). However, in diversify/alg/algs/diversify.py,Diversify.update_d()unpackes the data withI think it should be
to make it consistent.