(IMPORTANT): sawp input 4&5 for CNN model (need to check)

(IMPORTANT): sawp input 4&5 for CNN model (need to check) #16