Flock1:

(0): Conv2d(8, 1, kernel_size=(1, 1), stride=(1, 1))


This input layer requires 8 band input. as
Conv2d(in_channels, out_channels, kernel_size) which seems to me the bug so modify the original network to have input layer for 1 band and then extract the encoder.