add imagenet and cifar10 training network definition