Code stuck at testing

[gSingh-maker]

Latest issue : no change in test.py, except manually coding the best_file_name. No change in test.sh.

Below is the earlier issues I faced, probably we can ignore them, if we have the solution for above.:
downloaded the data_depth_selection, unzipped it, and then ran the following command :

The o/p is :

Ignore the line numbers. I use them inside test.py for debugging.
I changed the default value for --input_type inside test.py from "rgb", to "depth".

If I keep the --input_type to '"rgb" as default, I get the following o/p :

Could it be some issue with channel ? I see that depth_images in KItti dataset are 16 bit-depth (acc. to my understanding, that is equal to 2 channels), and rgb are 4 bit-depth (3 channels), whereas in test.py, its channel_in=1 for depth and 4 for rgb.

For input_type = 'rgb', channel_in=3, the o/p is:

I used the solution, pasted below, from this website https://discuss.pytorch.org/t/tensor-size-mismatch/31897/2

In test.py, for input_type = 'depth', channel_in=2
In test.sh, I changed the removed the highlighted part :

to the following

O/p :