So what I understand is summary can only work if the model has only 1 list of output and since mine output is 5 since it follows the structure of U-Net it can’t use summary, is it like that? Can you show how to use summary separately? Thank you for your help so far