PVT
PVT copied to clipboard
forward() missing 1 required positional argument: 'rel_pos'
When I wanted to train dataset, I met following problems.
What is "rel_pos"?
Perhaps shared_transformer.py file should add "rel_pos" parameter in Class SharedTransformer ?
Getting the same error.
My dataloader is extracting a single data input as size [batch_size, num_channels, num_points]
num_channels = 3 # for xyz channels
Not sure what rel_pos is or what its supposed to be. It isn't documented well in the example training scripts.
The trainloaders in main_part_seg.py
also seem to extract their batches similarly [batch_size, xyz_channel, num_points]
and pass that directly into the model without any extra argument.
Did you have any luck fixing this? @tomanick