Activation Extractor
latest
Contents:
Introduction
List of Models Included by Default
Python API:
Inferencers
Extractors
Model Specific Functions
Activation Extractor
Index
Edit on GitHub
Index
A
|
C
|
D
|
E
|
G
|
I
|
L
|
M
|
R
|
S
A
activation_extractor.model_functions.default_hooked_layers
module
activation_extractor.model_functions.embedding_to_numpy
module
activation_extractor.model_functions.inference_funs
module
activation_extractor.model_functions.load_models
module
activation_extractor.model_functions.model_types
module
activation_extractor.model_functions.tokenize_funs
module
C
clear_all_hooks() (activation_extractor.extractors.intermediateExtractorBase.IntermediateExtractorBase method)
create_hook() (activation_extractor.extractors.intermediateExtractorBase.IntermediateExtractorBase method)
D
define_inference_function() (in module activation_extractor.model_functions.inference_funs)
define_tokenize_function() (in module activation_extractor.model_functions.tokenize_funs)
detach_hooks() (activation_extractor.extractors.intermediateExtractorBase.IntermediateExtractorBase method)
E
emb_reformatting() (activation_extractor.extractors.intermediateExtractor.IntermediateExtractor method)
embedding_to_numpy() (in module activation_extractor.model_functions.embedding_to_numpy)
G
get_layers_to_hook() (in module activation_extractor.model_functions.default_hooked_layers)
get_outputs() (activation_extractor.extractors.intermediateExtractorBase.IntermediateExtractorBase method)
gpu_to_cpu() (activation_extractor.extractors.intermediateExtractor.IntermediateExtractor method)
I
IntermediateExtractor (class in activation_extractor.extractors.intermediateExtractor)
IntermediateExtractorBase (class in activation_extractor.extractors.intermediateExtractorBase)
L
load_model() (in module activation_extractor.model_functions.load_models)
load_tokenizer() (in module activation_extractor.model_functions.load_models)
M
module
activation_extractor.model_functions.default_hooked_layers
activation_extractor.model_functions.embedding_to_numpy
activation_extractor.model_functions.inference_funs
activation_extractor.model_functions.load_models
activation_extractor.model_functions.model_types
activation_extractor.model_functions.tokenize_funs
R
register_hooks() (activation_extractor.extractors.intermediateExtractorBase.IntermediateExtractorBase method)
S
save_outputs() (activation_extractor.extractors.intermediateExtractor.IntermediateExtractor method)