adding training module for CNN