HenryT:

File "/home/VCNN/scripts/vcnnmodel.py", line 167, in disc_loss
    rs_add = torch.norm(rs_center-6*conf['c0'],2,1,keepdim=True)   



It is a complicated function. I would try to debug from here, as it is a line getting you into the trouble. As I already mention, the simplest first try could be just to print the shape of the tensor, which is the input in this function.