A question about cmmd.py #33
Description
Maybe in cmmd function, threre should be "s_label = torch.zeros(32, args.nc).scatter_(1, s_label.data, 1)" rather than "s_label = torch.zeros(32, 31).scatter_(1, s_label.data, 1)"? Cause when I reviewed the code using "args.nc = 32", I got error like "RuntimeError: Invalid index in scatter". And when I change it and other relevant parts to "args.nc", it works.