TF-Keras

TF-Keras (legacy)

tf-keras는 Keras 2.x 버전에 붙여진 이름이에요. 지금은 여기 별도의 GitHub 저장소로 호스팅되고 있습니다. 레거시 프레임워크임에도 Hub에 4.5k+ 모델이 여전히 호스팅되어 있어요.

출처: 문서

본문

TF-Keras (legacy)

tf-keras는 Keras 2.x 버전에 붙여진 이름입니다. 지금은 여기 별도의 GitHub 저장소로 호스팅되고 있어요. 레거시 프레임워크이지만 여전히 4.5k+ 모델이 Hub에 호스팅되어 있습니다. 이 모델들은 huggingface_hub 라이브러리로 불러올 수 있어요. 머신에 tf-keras 또는 keras<3.x가 설치되어 있어야 합니다.

Keras 3.x 지원에 관심이 있다면 이 가이드를 확인하세요.

설치 후에는 from_pretrained_keras 메서드만 사용해 Hub에서 모델을 불러오면 됩니다. from_pretrained_keras에 대해 더 알아보려면 여기를 읽어보세요.

from huggingface_hub import from_pretrained_keras

model = from_pretrained_keras("keras-io/mobile-vit-xxs")
prediction = model.predict(image)
prediction = tf.squeeze(tf.round(prediction))
print(f'The image is a {classes[(np.argmax(prediction))]}!')

# The image is a sunflower!

내 tf-keras 모델을 Hub에 호스팅할 수도 있어요. 다만 tf-keras는 레거시 프레임워크라는 점을 기억하세요. 최대한 많은 사용자에게 도달하려면 Keras 3.x로 모델을 만들고 위에서 설명한 대로 네이티브하게 공유하는 것을 권장합니다. tf-keras 모델 업로드의 자세한 내용은 push_to_hub_keras 문서를 확인하세요.

from huggingface_hub import push_to_hub_keras

push_to_hub_keras(model,
    "your-username/your-model-name",
    "your-tensorboard-log-directory",
    tags = ["object-detection", "some_other_tag"],
    **model_save_kwargs,
)

추가 리소스 (Additional resources)

• GitHub 저장소
• 블로그 글 Putting Keras on 🤗 Hub for Collaborative Training and Reproducibility (2022년 4월)

더 알아보기 (Learn more)

tf-keras(Keras 2.x) 모델은 from_pretrained_keras("repo_id")로, 공유는 push_to_hub_keras(...)로 할 수 있어요. 다만 레거시 프레임워크이므로 새 모델은 Keras 3.x로 만들고 네이티브하게 공유하는 걸 권장합니다. 사용 전에 tf-keras 또는 keras<3.x 설치가 필요해요.