Index

_

__init__() (medtokenizers.AttnBlock method) (medtokenizers.BaseQuantizer method) (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.Decoder method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.Encoder method) (medtokenizers.FSQuantizer method) (medtokenizers.LFQuantizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.Normalize method) (medtokenizers.RAETokenizer method) (medtokenizers.ResidualFSQuantizer method) (medtokenizers.SpatialCompressor method) (medtokenizers.SpatialDecompressor method) (medtokenizers.TiTokTokenizer method) (medtokenizers.TokenizerEvaluator method) (medtokenizers.training.Checkpoint method) (medtokenizers.training.EarlyStopping method) (medtokenizers.training.Logger method) (medtokenizers.training.LossConfig method) (medtokenizers.training.LRScheduler method) (medtokenizers.training.Trainer method) (medtokenizers.VectorQuantizer method)

A

add_module() (medtokenizers.AttnBlock method) (medtokenizers.BaseQuantizer method) (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.Decoder method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.Encoder method) (medtokenizers.FSQuantizer method) (medtokenizers.LFQuantizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.ResidualFSQuantizer method) (medtokenizers.SpatialCompressor method) (medtokenizers.SpatialDecompressor method) (medtokenizers.TiTokTokenizer method) (medtokenizers.VectorQuantizer method)

apply() (medtokenizers.AttnBlock method) (medtokenizers.BaseQuantizer method) (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.Decoder method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.Encoder method) (medtokenizers.FSQuantizer method) (medtokenizers.LFQuantizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.ResidualFSQuantizer method) (medtokenizers.SpatialCompressor method) (medtokenizers.SpatialDecompressor method) (medtokenizers.TiTokTokenizer method) (medtokenizers.VectorQuantizer method) AttnBlock (class in medtokenizers)

B

BaseQuantizer (class in medtokenizers) BaseTokenizer (class in medtokenizers) bfloat16() (medtokenizers.AttnBlock method) (medtokenizers.BaseQuantizer method) (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.Decoder method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.Encoder method) (medtokenizers.FSQuantizer method) (medtokenizers.LFQuantizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.ResidualFSQuantizer method) (medtokenizers.SpatialCompressor method) (medtokenizers.SpatialDecompressor method) (medtokenizers.TiTokTokenizer method) (medtokenizers.VectorQuantizer method)

bound() (medtokenizers.FSQuantizer method) buffers() (medtokenizers.AttnBlock method) (medtokenizers.BaseQuantizer method) (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.Decoder method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.Encoder method) (medtokenizers.FSQuantizer method) (medtokenizers.LFQuantizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.ResidualFSQuantizer method) (medtokenizers.SpatialCompressor method) (medtokenizers.SpatialDecompressor method) (medtokenizers.TiTokTokenizer method) (medtokenizers.VectorQuantizer method)

C

call_super_init (medtokenizers.AttnBlock attribute) (medtokenizers.BaseQuantizer attribute) (medtokenizers.BaseTokenizer attribute) (medtokenizers.ContinuousTokenizer attribute) (medtokenizers.Decoder attribute) (medtokenizers.DiscreteTokenizer attribute) (medtokenizers.Encoder attribute) (medtokenizers.FSQuantizer attribute) (medtokenizers.LFQuantizer attribute) (medtokenizers.MAISITokenizer attribute) (medtokenizers.RAETokenizer attribute) (medtokenizers.ResidualFSQuantizer attribute) (medtokenizers.SpatialCompressor attribute) (medtokenizers.SpatialDecompressor attribute) (medtokenizers.TiTokTokenizer attribute) (medtokenizers.VectorQuantizer attribute) Callback (class in medtokenizers.training) Checkpoint (class in medtokenizers.training) children() (medtokenizers.AttnBlock method) (medtokenizers.BaseQuantizer method) (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.Decoder method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.Encoder method) (medtokenizers.FSQuantizer method) (medtokenizers.LFQuantizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.ResidualFSQuantizer method) (medtokenizers.SpatialCompressor method) (medtokenizers.SpatialDecompressor method) (medtokenizers.TiTokTokenizer method) (medtokenizers.VectorQuantizer method) clear_metric_caches() (in module medtokenizers) codes_to_indices() (medtokenizers.FSQuantizer method) compile() (medtokenizers.AttnBlock method) (medtokenizers.BaseQuantizer method) (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.Decoder method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.Encoder method) (medtokenizers.FSQuantizer method) (medtokenizers.LFQuantizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.ResidualFSQuantizer method) (medtokenizers.SpatialCompressor method) (medtokenizers.SpatialDecompressor method) (medtokenizers.TiTokTokenizer method) (medtokenizers.VectorQuantizer method)

compute_codebook_usage() (in module medtokenizers) compute_lpips() (in module medtokenizers) compute_mae() (in module medtokenizers) compute_mse() (in module medtokenizers) compute_perplexity() (in module medtokenizers) compute_psnr() (in module medtokenizers) compute_ssim() (in module medtokenizers) compute_stage_weights() (medtokenizers.training.LossConfig method) config (medtokenizers.BaseTokenizer attribute) (medtokenizers.ContinuousTokenizer attribute) (medtokenizers.DiscreteTokenizer attribute) (medtokenizers.MAISITokenizer attribute) (medtokenizers.RAETokenizer attribute) (medtokenizers.TiTokTokenizer attribute) config_name (medtokenizers.BaseTokenizer attribute) (medtokenizers.ContinuousTokenizer attribute) (medtokenizers.DiscreteTokenizer attribute) (medtokenizers.MAISITokenizer attribute) (medtokenizers.RAETokenizer attribute) (medtokenizers.TiTokTokenizer attribute) ContinuousTokenizer (class in medtokenizers) cpu() (medtokenizers.AttnBlock method) (medtokenizers.BaseQuantizer method) (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.Decoder method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.Encoder method) (medtokenizers.FSQuantizer method) (medtokenizers.LFQuantizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.ResidualFSQuantizer method) (medtokenizers.SpatialCompressor method) (medtokenizers.SpatialDecompressor method) (medtokenizers.TiTokTokenizer method) (medtokenizers.VectorQuantizer method) cuda() (medtokenizers.AttnBlock method) (medtokenizers.BaseQuantizer method) (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.Decoder method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.Encoder method) (medtokenizers.FSQuantizer method) (medtokenizers.LFQuantizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.ResidualFSQuantizer method) (medtokenizers.SpatialCompressor method) (medtokenizers.SpatialDecompressor method) (medtokenizers.TiTokTokenizer method) (medtokenizers.VectorQuantizer method)

D

decode() (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.TiTokTokenizer method) decode_batch() (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.TiTokTokenizer method) Decoder (class in medtokenizers) DEFAULT_CONFIG (medtokenizers.MAISITokenizer attribute) detokenize() (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.TiTokTokenizer method) DiscreteTokenizer (class in medtokenizers) double() (medtokenizers.AttnBlock method) (medtokenizers.BaseQuantizer method) (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.Decoder method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.Encoder method) (medtokenizers.FSQuantizer method) (medtokenizers.LFQuantizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.ResidualFSQuantizer method) (medtokenizers.SpatialCompressor method) (medtokenizers.SpatialDecompressor method) (medtokenizers.TiTokTokenizer method) (medtokenizers.VectorQuantizer method)

dtype (medtokenizers.FSQuantizer attribute) (medtokenizers.LFQuantizer attribute) (medtokenizers.ResidualFSQuantizer attribute) (medtokenizers.VectorQuantizer attribute) dump_patches (medtokenizers.AttnBlock attribute) (medtokenizers.BaseQuantizer attribute) (medtokenizers.BaseTokenizer attribute) (medtokenizers.ContinuousTokenizer attribute) (medtokenizers.Decoder attribute) (medtokenizers.DiscreteTokenizer attribute) (medtokenizers.Encoder attribute) (medtokenizers.FSQuantizer attribute) (medtokenizers.LFQuantizer attribute) (medtokenizers.MAISITokenizer attribute) (medtokenizers.RAETokenizer attribute) (medtokenizers.ResidualFSQuantizer attribute) (medtokenizers.SpatialCompressor attribute) (medtokenizers.SpatialDecompressor attribute) (medtokenizers.TiTokTokenizer attribute) (medtokenizers.VectorQuantizer attribute)

E

EarlyStopping (class in medtokenizers.training) encode() (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.TiTokTokenizer method) encode_batch() (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.TiTokTokenizer method) Encoder (class in medtokenizers) end_training() (medtokenizers.training.Trainer method) eval() (medtokenizers.AttnBlock method) (medtokenizers.BaseQuantizer method) (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.Decoder method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.Encoder method) (medtokenizers.FSQuantizer method) (medtokenizers.LFQuantizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.ResidualFSQuantizer method) (medtokenizers.SpatialCompressor method) (medtokenizers.SpatialDecompressor method) (medtokenizers.TiTokTokenizer method) (medtokenizers.VectorQuantizer method)

evaluate() (medtokenizers.TokenizerEvaluator method) evaluate_batch() (medtokenizers.TokenizerEvaluator method) extra_repr() (medtokenizers.AttnBlock method) (medtokenizers.BaseQuantizer method) (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.Decoder method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.Encoder method) (medtokenizers.FSQuantizer method) (medtokenizers.LFQuantizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.ResidualFSQuantizer method) (medtokenizers.SpatialCompressor method) (medtokenizers.SpatialDecompressor method) (medtokenizers.TiTokTokenizer method) (medtokenizers.VectorQuantizer method)

F

fit() (medtokenizers.training.Trainer method) float() (medtokenizers.AttnBlock method) (medtokenizers.BaseQuantizer method) (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.Decoder method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.Encoder method) (medtokenizers.FSQuantizer method) (medtokenizers.LFQuantizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.ResidualFSQuantizer method) (medtokenizers.SpatialCompressor method) (medtokenizers.SpatialDecompressor method) (medtokenizers.TiTokTokenizer method) (medtokenizers.VectorQuantizer method) forward() (medtokenizers.AttnBlock method) (medtokenizers.BaseQuantizer method) (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.Decoder method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.Encoder method) (medtokenizers.FSQuantizer method) (medtokenizers.LFQuantizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.ResidualFSQuantizer method) (medtokenizers.SpatialCompressor method) (medtokenizers.SpatialDecompressor method) (medtokenizers.TiTokTokenizer method) (medtokenizers.VectorQuantizer method)

from_pretrained() (medtokenizers.BaseTokenizer class method) (medtokenizers.ContinuousTokenizer class method) (medtokenizers.DiscreteTokenizer class method) (medtokenizers.MAISITokenizer class method) (medtokenizers.RAETokenizer class method) (medtokenizers.TiTokTokenizer class method) from_pretrained_encoder_decoder() (medtokenizers.BaseTokenizer class method) (medtokenizers.ContinuousTokenizer class method) (medtokenizers.DiscreteTokenizer class method) (medtokenizers.MAISITokenizer class method) (medtokenizers.RAETokenizer class method) (medtokenizers.TiTokTokenizer class method) FSQuantizer (class in medtokenizers)

G

get_buffer() (medtokenizers.AttnBlock method) (medtokenizers.BaseQuantizer method) (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.Decoder method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.Encoder method) (medtokenizers.FSQuantizer method) (medtokenizers.LFQuantizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.ResidualFSQuantizer method) (medtokenizers.SpatialCompressor method) (medtokenizers.SpatialDecompressor method) (medtokenizers.TiTokTokenizer method) (medtokenizers.VectorQuantizer method) get_codebook_entry() (medtokenizers.VectorQuantizer method) get_codebook_size() (medtokenizers.BaseQuantizer method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.FSQuantizer method) (medtokenizers.LFQuantizer method) (medtokenizers.ResidualFSQuantizer method) (medtokenizers.VectorQuantizer method) get_codebook_usage() (medtokenizers.VectorQuantizer method) get_config() (in module medtokenizers) get_extra_state() (medtokenizers.AttnBlock method) (medtokenizers.BaseQuantizer method) (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.Decoder method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.Encoder method) (medtokenizers.FSQuantizer method) (medtokenizers.LFQuantizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.ResidualFSQuantizer method) (medtokenizers.SpatialCompressor method) (medtokenizers.SpatialDecompressor method) (medtokenizers.TiTokTokenizer method) (medtokenizers.VectorQuantizer method)

get_latent_shape() (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.TiTokTokenizer method) get_parameter() (medtokenizers.AttnBlock method) (medtokenizers.BaseQuantizer method) (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.Decoder method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.Encoder method) (medtokenizers.FSQuantizer method) (medtokenizers.LFQuantizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.ResidualFSQuantizer method) (medtokenizers.SpatialCompressor method) (medtokenizers.SpatialDecompressor method) (medtokenizers.TiTokTokenizer method) (medtokenizers.VectorQuantizer method) get_submodule() (medtokenizers.AttnBlock method) (medtokenizers.BaseQuantizer method) (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.Decoder method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.Encoder method) (medtokenizers.FSQuantizer method) (medtokenizers.LFQuantizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.ResidualFSQuantizer method) (medtokenizers.SpatialCompressor method) (medtokenizers.SpatialDecompressor method) (medtokenizers.TiTokTokenizer method) (medtokenizers.VectorQuantizer method) get_training_config() (medtokenizers.MAISITokenizer static method) gram_weight (medtokenizers.training.LossConfig attribute)

H

half() (medtokenizers.AttnBlock method) (medtokenizers.BaseQuantizer method) (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.Decoder method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.Encoder method) (medtokenizers.FSQuantizer method) (medtokenizers.LFQuantizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.ResidualFSQuantizer method) (medtokenizers.SpatialCompressor method) (medtokenizers.SpatialDecompressor method) (medtokenizers.TiTokTokenizer method) (medtokenizers.VectorQuantizer method)

I

implicit_codebook (medtokenizers.FSQuantizer property) indices_to_codes() (medtokenizers.BaseQuantizer method) (medtokenizers.FSQuantizer method) (medtokenizers.LFQuantizer method) (medtokenizers.ResidualFSQuantizer method) (medtokenizers.TiTokTokenizer method) (medtokenizers.VectorQuantizer method) inference_mode() (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.TiTokTokenizer method) init_trackers() (medtokenizers.training.Trainer method) ipu() (medtokenizers.AttnBlock method) (medtokenizers.BaseQuantizer method) (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.Decoder method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.Encoder method) (medtokenizers.FSQuantizer method) (medtokenizers.LFQuantizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.ResidualFSQuantizer method) (medtokenizers.SpatialCompressor method) (medtokenizers.SpatialDecompressor method) (medtokenizers.TiTokTokenizer method) (medtokenizers.VectorQuantizer method)

L

l1_weight (medtokenizers.training.LossConfig attribute) laplacian_start_epoch (medtokenizers.training.LossConfig attribute) laplacian_weight (medtokenizers.training.LossConfig attribute) LFQuantizer (class in medtokenizers) load_checkpoint() (medtokenizers.training.Trainer method) load_encoder_decoder_weights() (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.TiTokTokenizer method) load_indices() (in module medtokenizers) load_latents() (in module medtokenizers) load_results() (medtokenizers.TokenizerEvaluator static method) load_state_dict() (medtokenizers.AttnBlock method) (medtokenizers.BaseQuantizer method) (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.Decoder method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.Encoder method) (medtokenizers.FSQuantizer method) (medtokenizers.LFQuantizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.ResidualFSQuantizer method) (medtokenizers.SpatialCompressor method) (medtokenizers.SpatialDecompressor method) (medtokenizers.TiTokTokenizer method) (medtokenizers.VectorQuantizer method)

load_tokenizer() (in module medtokenizers) Logger (class in medtokenizers.training) LossConfig (class in medtokenizers.training) LRScheduler (class in medtokenizers.training)

M

MAISITokenizer (class in medtokenizers) modules() (medtokenizers.AttnBlock method) (medtokenizers.BaseQuantizer method) (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.Decoder method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.Encoder method) (medtokenizers.FSQuantizer method) (medtokenizers.LFQuantizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.ResidualFSQuantizer method) (medtokenizers.SpatialCompressor method) (medtokenizers.SpatialDecompressor method) (medtokenizers.TiTokTokenizer method) (medtokenizers.VectorQuantizer method)

mtia() (medtokenizers.AttnBlock method) (medtokenizers.BaseQuantizer method) (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.Decoder method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.Encoder method) (medtokenizers.FSQuantizer method) (medtokenizers.LFQuantizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.ResidualFSQuantizer method) (medtokenizers.SpatialCompressor method) (medtokenizers.SpatialDecompressor method) (medtokenizers.TiTokTokenizer method) (medtokenizers.VectorQuantizer method)

N

named_buffers() (medtokenizers.AttnBlock method) (medtokenizers.BaseQuantizer method) (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.Decoder method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.Encoder method) (medtokenizers.FSQuantizer method) (medtokenizers.LFQuantizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.ResidualFSQuantizer method) (medtokenizers.SpatialCompressor method) (medtokenizers.SpatialDecompressor method) (medtokenizers.TiTokTokenizer method) (medtokenizers.VectorQuantizer method) named_children() (medtokenizers.AttnBlock method) (medtokenizers.BaseQuantizer method) (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.Decoder method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.Encoder method) (medtokenizers.FSQuantizer method) (medtokenizers.LFQuantizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.ResidualFSQuantizer method) (medtokenizers.SpatialCompressor method) (medtokenizers.SpatialDecompressor method) (medtokenizers.TiTokTokenizer method) (medtokenizers.VectorQuantizer method) named_modules() (medtokenizers.AttnBlock method) (medtokenizers.BaseQuantizer method) (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.Decoder method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.Encoder method) (medtokenizers.FSQuantizer method) (medtokenizers.LFQuantizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.ResidualFSQuantizer method) (medtokenizers.SpatialCompressor method) (medtokenizers.SpatialDecompressor method) (medtokenizers.TiTokTokenizer method) (medtokenizers.VectorQuantizer method)

named_parameters() (medtokenizers.AttnBlock method) (medtokenizers.BaseQuantizer method) (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.Decoder method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.Encoder method) (medtokenizers.FSQuantizer method) (medtokenizers.LFQuantizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.ResidualFSQuantizer method) (medtokenizers.SpatialCompressor method) (medtokenizers.SpatialDecompressor method) (medtokenizers.TiTokTokenizer method) (medtokenizers.VectorQuantizer method) Normalize (class in medtokenizers) num_parameters() (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.TiTokTokenizer method)

O

on_batch_begin() (medtokenizers.training.Callback method) (medtokenizers.training.Checkpoint method) (medtokenizers.training.EarlyStopping method) (medtokenizers.training.Logger method) (medtokenizers.training.LRScheduler method) on_batch_end() (medtokenizers.training.Callback method) (medtokenizers.training.Checkpoint method) (medtokenizers.training.EarlyStopping method) (medtokenizers.training.Logger method) (medtokenizers.training.LRScheduler method) on_epoch_begin() (medtokenizers.training.Callback method) (medtokenizers.training.Checkpoint method) (medtokenizers.training.EarlyStopping method) (medtokenizers.training.Logger method) (medtokenizers.training.LRScheduler method) on_epoch_end() (medtokenizers.training.Callback method) (medtokenizers.training.Checkpoint method) (medtokenizers.training.EarlyStopping method) (medtokenizers.training.Logger method) (medtokenizers.training.LRScheduler method)

on_train_begin() (medtokenizers.training.Callback method) (medtokenizers.training.Checkpoint method) (medtokenizers.training.EarlyStopping method) (medtokenizers.training.Logger method) (medtokenizers.training.LRScheduler method) on_train_end() (medtokenizers.training.Callback method) (medtokenizers.training.Checkpoint method) (medtokenizers.training.EarlyStopping method) (medtokenizers.training.Logger method) (medtokenizers.training.LRScheduler method) on_validation_begin() (medtokenizers.training.Callback method) (medtokenizers.training.Checkpoint method) (medtokenizers.training.EarlyStopping method) (medtokenizers.training.Logger method) (medtokenizers.training.LRScheduler method) on_validation_end() (medtokenizers.training.Callback method) (medtokenizers.training.Checkpoint method) (medtokenizers.training.EarlyStopping method) (medtokenizers.training.Logger method) (medtokenizers.training.LRScheduler method)

P

pad_divisible() (in module medtokenizers) parameters() (medtokenizers.AttnBlock method) (medtokenizers.BaseQuantizer method) (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.Decoder method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.Encoder method) (medtokenizers.FSQuantizer method) (medtokenizers.LFQuantizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.ResidualFSQuantizer method) (medtokenizers.SpatialCompressor method) (medtokenizers.SpatialDecompressor method) (medtokenizers.TiTokTokenizer method) (medtokenizers.VectorQuantizer method)

percentile_normalize() (in module medtokenizers) postprocess_from_maisi() (in module medtokenizers) preprocess_for_maisi() (in module medtokenizers) print_results() (medtokenizers.TokenizerEvaluator method) push_to_hub() (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.TiTokTokenizer method)

Q

quant_weight (medtokenizers.training.LossConfig attribute)

quantize() (medtokenizers.FSQuantizer method)

R

RAETokenizer (class in medtokenizers) reconstruct() (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.TiTokTokenizer method) register_backward_hook() (medtokenizers.AttnBlock method) (medtokenizers.BaseQuantizer method) (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.Decoder method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.Encoder method) (medtokenizers.FSQuantizer method) (medtokenizers.LFQuantizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.ResidualFSQuantizer method) (medtokenizers.SpatialCompressor method) (medtokenizers.SpatialDecompressor method) (medtokenizers.TiTokTokenizer method) (medtokenizers.VectorQuantizer method) register_buffer() (medtokenizers.AttnBlock method) (medtokenizers.BaseQuantizer method) (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.Decoder method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.Encoder method) (medtokenizers.FSQuantizer method) (medtokenizers.LFQuantizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.ResidualFSQuantizer method) (medtokenizers.SpatialCompressor method) (medtokenizers.SpatialDecompressor method) (medtokenizers.TiTokTokenizer method) (medtokenizers.VectorQuantizer method) register_forward_hook() (medtokenizers.AttnBlock method) (medtokenizers.BaseQuantizer method) (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.Decoder method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.Encoder method) (medtokenizers.FSQuantizer method) (medtokenizers.LFQuantizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.ResidualFSQuantizer method) (medtokenizers.SpatialCompressor method) (medtokenizers.SpatialDecompressor method) (medtokenizers.TiTokTokenizer method) (medtokenizers.VectorQuantizer method) register_forward_pre_hook() (medtokenizers.AttnBlock method) (medtokenizers.BaseQuantizer method) (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.Decoder method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.Encoder method) (medtokenizers.FSQuantizer method) (medtokenizers.LFQuantizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.ResidualFSQuantizer method) (medtokenizers.SpatialCompressor method) (medtokenizers.SpatialDecompressor method) (medtokenizers.TiTokTokenizer method) (medtokenizers.VectorQuantizer method) register_full_backward_hook() (medtokenizers.AttnBlock method) (medtokenizers.BaseQuantizer method) (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.Decoder method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.Encoder method) (medtokenizers.FSQuantizer method) (medtokenizers.LFQuantizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.ResidualFSQuantizer method) (medtokenizers.SpatialCompressor method) (medtokenizers.SpatialDecompressor method) (medtokenizers.TiTokTokenizer method) (medtokenizers.VectorQuantizer method) register_full_backward_pre_hook() (medtokenizers.AttnBlock method) (medtokenizers.BaseQuantizer method) (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.Decoder method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.Encoder method) (medtokenizers.FSQuantizer method) (medtokenizers.LFQuantizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.ResidualFSQuantizer method) (medtokenizers.SpatialCompressor method) (medtokenizers.SpatialDecompressor method) (medtokenizers.TiTokTokenizer method) (medtokenizers.VectorQuantizer method) register_load_state_dict_post_hook() (medtokenizers.AttnBlock method) (medtokenizers.BaseQuantizer method) (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.Decoder method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.Encoder method) (medtokenizers.FSQuantizer method) (medtokenizers.LFQuantizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.ResidualFSQuantizer method) (medtokenizers.SpatialCompressor method) (medtokenizers.SpatialDecompressor method) (medtokenizers.TiTokTokenizer method) (medtokenizers.VectorQuantizer method)

register_load_state_dict_pre_hook() (medtokenizers.AttnBlock method) (medtokenizers.BaseQuantizer method) (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.Decoder method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.Encoder method) (medtokenizers.FSQuantizer method) (medtokenizers.LFQuantizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.ResidualFSQuantizer method) (medtokenizers.SpatialCompressor method) (medtokenizers.SpatialDecompressor method) (medtokenizers.TiTokTokenizer method) (medtokenizers.VectorQuantizer method) register_module() (medtokenizers.AttnBlock method) (medtokenizers.BaseQuantizer method) (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.Decoder method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.Encoder method) (medtokenizers.FSQuantizer method) (medtokenizers.LFQuantizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.ResidualFSQuantizer method) (medtokenizers.SpatialCompressor method) (medtokenizers.SpatialDecompressor method) (medtokenizers.TiTokTokenizer method) (medtokenizers.VectorQuantizer method) register_parameter() (medtokenizers.AttnBlock method) (medtokenizers.BaseQuantizer method) (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.Decoder method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.Encoder method) (medtokenizers.FSQuantizer method) (medtokenizers.LFQuantizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.ResidualFSQuantizer method) (medtokenizers.SpatialCompressor method) (medtokenizers.SpatialDecompressor method) (medtokenizers.TiTokTokenizer method) (medtokenizers.VectorQuantizer method) register_state_dict_post_hook() (medtokenizers.AttnBlock method) (medtokenizers.BaseQuantizer method) (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.Decoder method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.Encoder method) (medtokenizers.FSQuantizer method) (medtokenizers.LFQuantizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.ResidualFSQuantizer method) (medtokenizers.SpatialCompressor method) (medtokenizers.SpatialDecompressor method) (medtokenizers.TiTokTokenizer method) (medtokenizers.VectorQuantizer method) register_state_dict_pre_hook() (medtokenizers.AttnBlock method) (medtokenizers.BaseQuantizer method) (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.Decoder method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.Encoder method) (medtokenizers.FSQuantizer method) (medtokenizers.LFQuantizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.ResidualFSQuantizer method) (medtokenizers.SpatialCompressor method) (medtokenizers.SpatialDecompressor method) (medtokenizers.TiTokTokenizer method) (medtokenizers.VectorQuantizer method) requires_grad_() (medtokenizers.AttnBlock method) (medtokenizers.BaseQuantizer method) (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.Decoder method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.Encoder method) (medtokenizers.FSQuantizer method) (medtokenizers.LFQuantizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.ResidualFSQuantizer method) (medtokenizers.SpatialCompressor method) (medtokenizers.SpatialDecompressor method) (medtokenizers.TiTokTokenizer method) (medtokenizers.VectorQuantizer method) resample_to_spacing() (in module medtokenizers) ResidualFSQuantizer (class in medtokenizers)

S

save_checkpoint() (medtokenizers.training.Trainer method) save_indices() (in module medtokenizers) save_latents() (in module medtokenizers) save_pretrained() (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.TiTokTokenizer method) save_results() (medtokenizers.TokenizerEvaluator static method) scale (medtokenizers.AttnBlock attribute) set_extra_state() (medtokenizers.AttnBlock method) (medtokenizers.BaseQuantizer method) (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.Decoder method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.Encoder method) (medtokenizers.FSQuantizer method) (medtokenizers.LFQuantizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.ResidualFSQuantizer method) (medtokenizers.SpatialCompressor method) (medtokenizers.SpatialDecompressor method) (medtokenizers.TiTokTokenizer method) (medtokenizers.VectorQuantizer method) set_submodule() (medtokenizers.AttnBlock method) (medtokenizers.BaseQuantizer method) (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.Decoder method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.Encoder method) (medtokenizers.FSQuantizer method) (medtokenizers.LFQuantizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.ResidualFSQuantizer method) (medtokenizers.SpatialCompressor method) (medtokenizers.SpatialDecompressor method) (medtokenizers.TiTokTokenizer method) (medtokenizers.VectorQuantizer method)

share_memory() (medtokenizers.AttnBlock method) (medtokenizers.BaseQuantizer method) (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.Decoder method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.Encoder method) (medtokenizers.FSQuantizer method) (medtokenizers.LFQuantizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.ResidualFSQuantizer method) (medtokenizers.SpatialCompressor method) (medtokenizers.SpatialDecompressor method) (medtokenizers.TiTokTokenizer method) (medtokenizers.VectorQuantizer method) SpatialCompressor (class in medtokenizers) SpatialDecompressor (class in medtokenizers) stage1_epochs (medtokenizers.training.LossConfig attribute) stage1_gram_weight (medtokenizers.training.LossConfig attribute) stage1_vgg_weight (medtokenizers.training.LossConfig attribute) stage2_gram_weight (medtokenizers.training.LossConfig attribute) stage2_vgg_weight (medtokenizers.training.LossConfig attribute) stage2_warmup_epochs (medtokenizers.training.LossConfig attribute) state_dict() (medtokenizers.AttnBlock method) (medtokenizers.BaseQuantizer method) (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.Decoder method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.Encoder method) (medtokenizers.FSQuantizer method) (medtokenizers.LFQuantizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.ResidualFSQuantizer method) (medtokenizers.SpatialCompressor method) (medtokenizers.SpatialDecompressor method) (medtokenizers.TiTokTokenizer method) (medtokenizers.VectorQuantizer method)

T

T_destination (medtokenizers.AttnBlock attribute) (medtokenizers.BaseQuantizer attribute) (medtokenizers.BaseTokenizer attribute) (medtokenizers.ContinuousTokenizer attribute) (medtokenizers.Decoder attribute) (medtokenizers.DiscreteTokenizer attribute) (medtokenizers.Encoder attribute) (medtokenizers.FSQuantizer attribute) (medtokenizers.LFQuantizer attribute) (medtokenizers.MAISITokenizer attribute) (medtokenizers.RAETokenizer attribute) (medtokenizers.ResidualFSQuantizer attribute) (medtokenizers.SpatialCompressor attribute) (medtokenizers.SpatialDecompressor attribute) (medtokenizers.TiTokTokenizer attribute) (medtokenizers.VectorQuantizer attribute) TiTokTokenizer (class in medtokenizers) to() (medtokenizers.AttnBlock method) (medtokenizers.BaseQuantizer method) (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.Decoder method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.Encoder method) (medtokenizers.FSQuantizer method) (medtokenizers.LFQuantizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.ResidualFSQuantizer method) (medtokenizers.SpatialCompressor method) (medtokenizers.SpatialDecompressor method) (medtokenizers.TiTokTokenizer method) (medtokenizers.VectorQuantizer method) to_empty() (medtokenizers.AttnBlock method) (medtokenizers.BaseQuantizer method) (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.Decoder method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.Encoder method) (medtokenizers.FSQuantizer method) (medtokenizers.LFQuantizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.ResidualFSQuantizer method) (medtokenizers.SpatialCompressor method) (medtokenizers.SpatialDecompressor method) (medtokenizers.TiTokTokenizer method) (medtokenizers.VectorQuantizer method) tokenize() (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.TiTokTokenizer method)

TokenizerEvaluator (class in medtokenizers) train() (medtokenizers.AttnBlock method) (medtokenizers.BaseQuantizer method) (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.Decoder method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.Encoder method) (medtokenizers.FSQuantizer method) (medtokenizers.LFQuantizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.ResidualFSQuantizer method) (medtokenizers.SpatialCompressor method) (medtokenizers.SpatialDecompressor method) (medtokenizers.TiTokTokenizer method) (medtokenizers.VectorQuantizer method) train_step() (medtokenizers.training.Trainer method) Trainer (class in medtokenizers.training) training (medtokenizers.AttnBlock attribute) (medtokenizers.BaseQuantizer attribute) (medtokenizers.BaseTokenizer attribute) (medtokenizers.ContinuousTokenizer attribute) (medtokenizers.Decoder attribute) (medtokenizers.DiscreteTokenizer attribute) (medtokenizers.Encoder attribute) (medtokenizers.FSQuantizer attribute) (medtokenizers.LFQuantizer attribute) (medtokenizers.MAISITokenizer attribute) (medtokenizers.RAETokenizer attribute) (medtokenizers.ResidualFSQuantizer attribute) (medtokenizers.SpatialCompressor attribute) (medtokenizers.SpatialDecompressor attribute) (medtokenizers.TiTokTokenizer attribute) (medtokenizers.VectorQuantizer attribute) type() (medtokenizers.AttnBlock method) (medtokenizers.BaseQuantizer method) (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.Decoder method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.Encoder method) (medtokenizers.FSQuantizer method) (medtokenizers.LFQuantizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.ResidualFSQuantizer method) (medtokenizers.SpatialCompressor method) (medtokenizers.SpatialDecompressor method) (medtokenizers.TiTokTokenizer method) (medtokenizers.VectorQuantizer method)

U

unpad() (in module medtokenizers) uses_gram (medtokenizers.training.LossConfig property)

uses_laplacian (medtokenizers.training.LossConfig property) uses_vgg (medtokenizers.training.LossConfig property)

V

validate() (medtokenizers.training.Trainer method)

VectorQuantizer (class in medtokenizers) vgg_weight (medtokenizers.training.LossConfig attribute)

W

weights_name (medtokenizers.BaseTokenizer attribute) (medtokenizers.ContinuousTokenizer attribute) (medtokenizers.DiscreteTokenizer attribute) (medtokenizers.MAISITokenizer attribute) (medtokenizers.RAETokenizer attribute) (medtokenizers.TiTokTokenizer attribute)

X

xpu() (medtokenizers.AttnBlock method) (medtokenizers.BaseQuantizer method) (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.Decoder method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.Encoder method) (medtokenizers.FSQuantizer method) (medtokenizers.LFQuantizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.ResidualFSQuantizer method) (medtokenizers.SpatialCompressor method) (medtokenizers.SpatialDecompressor method) (medtokenizers.TiTokTokenizer method) (medtokenizers.VectorQuantizer method)

Z

zero_grad() (medtokenizers.AttnBlock method) (medtokenizers.BaseQuantizer method) (medtokenizers.BaseTokenizer method) (medtokenizers.ContinuousTokenizer method) (medtokenizers.Decoder method) (medtokenizers.DiscreteTokenizer method) (medtokenizers.Encoder method) (medtokenizers.FSQuantizer method) (medtokenizers.LFQuantizer method) (medtokenizers.MAISITokenizer method) (medtokenizers.RAETokenizer method) (medtokenizers.ResidualFSQuantizer method) (medtokenizers.SpatialCompressor method) (medtokenizers.SpatialDecompressor method) (medtokenizers.TiTokTokenizer method) (medtokenizers.VectorQuantizer method)