The parameters of the pre-trained model are incorrect

[hit-hangh]

Hello, I downloaded the DIC-Net-d pre-training model parameters from Baidu Cloud web disk.“model = DICNet_d(2, 2).to(device)”，the out_channel must be set 3，Otherwise, an error will be reported. And the result is the same as DIC-Net-s.I guess you got mixed up when you uploaded it.I want you to see if the same problem occurs