Closed l912237463 closed 2 years ago
Since the number of data channels is not 3, I cannot use the pre-trained model. How can I modify the code
You just need to change in_channels in your model. For using pretrained model, refer here.
in_channels
Since the number of data channels is not 3, I cannot use the pre-trained model. How can I modify the code