EGIInet
EGIInet copied to clipboard
Training code issues
Thank you for your work! But I can't find the code for the backward propagation. Also, style_loss
in train.py
is a tensor with 4 items, which makes an error in total_cd_pc += cd_pc_item
.