transformers pytorch tf safetensors convnext image-classification vision dataset:imagenet-21k dataset:imagenet-1k arxiv:2201.03545 license:apache-2.0 autotrain_compatible endpoints_compatible region:us