infer consists of 3 stages in paper,but there is only 2 stage in code

[Yes-Jumby]

[Yes-Jumby]

i add one stage in code. the img w and h should be divisible by 8 for two stage the img w and h should be divisible by 16 for three stage