peach6

Error while training on custom dataset

7

Hi, Great work. I am trying to train the model on my custom dataset but I am stuck on this error. File "train_PBAFN_stage1_fs.py", line 106, in concat = torch.cat([preserve_mask.cuda(),densepose,pose.cuda()],1) RuntimeError:...