Solved by lelouedec in post #14 
              
              
                The problem here is that the input you are passing to the discriminator doesn’t seem to be of the correct size. 
So without your loading function and printing the size of the tensor before going in the model then it is difficult to assess