Hi, @ptrblck I am writing the UNet code using PyTorch Lightning.
I am taking help from a repo 3
The unet and train.py is working fine. But at the time of test.py, it is showing the error (question title and description) is the same for me.
In my understanding, when, we are running test.py here and any images going through the pre-trained unet model here which requires n_channels!
But from the test.py we are not passing any arguments like n_channels because it is already defined! Most probably for that reason, the error is showing.
Any idea to solve the issue?
Updated:
Problem solved