From 7563e031bd20fb2e7173a06d95ea7c91d243a45f Mon Sep 17 00:00:00 2001 From: spencerwooo Date: Mon, 15 Apr 2024 16:49:23 +0800 Subject: [PATCH] Update README tips --- README.md | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/README.md b/README.md index fa834f2..cc739c9 100644 --- a/README.md +++ b/README.md @@ -3,10 +3,10 @@ # torch-featurelayer -Simple utility functions and wrappers for hooking onto layers within PyTorch models for feature extraction. +🧠 Simple utility functions and wrappers for hooking onto layers within PyTorch models for feature extraction. -| 🧠 | For a more sophisticated and complete implementation, check the official [`torch.fx`](https://pytorch.org/docs/stable/fx.html). | -| --- | :------------------------------------------------------------------------------------------------------------------------------ | +> [!TIP] +> This library is intended to be a simplified and well-documented implementation for extracting a PyTorch model's intermediate layer output(s). For a more sophisticated and complete implementation, either consider using [`torchvision.models.feature_extraction`](https://pytorch.org/vision/stable/feature_extraction.html), or check the official [`torch.fx`](https://pytorch.org/docs/stable/fx.html). ## Usage