Hi, thanks for the great repo. The 'Visual Representation' shows that kernel size used for Down Conv and Up Conv layers used is (2x2x1). However, it's (2x2x2) in the code. The latter one makes sense though. Could you please let me know if it's a typo in the figure?