different size between pretrain and finetune

[zhangdan0602]

Hi, thanks for your contribution. I have a question in the implementation process. For molcpba, firstly we use graphormer to pretrain on CQM4Mv2 , which in num_class is 1. Thus, the size is torch.Size([1, 1024]). However, when we finetune on the molpcba, the num_class is 128，size is torch.Size([128, 1024]). Thus their size is mismatch. How do we address it? Thanks.