Solved problem with Dropout when computing input/output sizes