Ah, of course, I tried this as well but replaced it again when trying with other Densenet variants. Unfortunately, I just forgot for the example, I just double-checked and it does not fix the original issue.