Open senliontec opened 1 month ago
The function is used for getting first conv, so it skips the loop when module is nn.Conv2d. You can refer to https://github.com/qubvel-org/segmentation_models.pytorch/blob/main/segmentation_models_pytorch/encoders/_utils.py.
In sigma_logit_unetpp.py:
line 165 - 170
the code is as following:
I guess the variable 'weight' is in the for loop, please check it and the following codes.
thank you ~