Question about the Classblock

[zkangkang0]

Hello, I've been reading this code recently. I have a problem that I don't understand.

In the model.py file, the part_classifier function defined in the three_view_net function. During the training, the structure of the Classblock in this function is as follows:

During the test, the structure of classblock is as follows:

I want to ask, how is this achieved?

[wtyhub]

@zkangkang0
This structure is achieved using these codes.
https://github.com/wtyhub/LPN/blob/main/test.py#L246-L254

[zkangkang0]

Thank you so much for your reply