add imagenet category wise trainer