Hide navigation sidebar

Hide table of contents sidebar

Toggle site navigation sidebar

Toggle table of contents sidebar

Getting started

First steps
Training, Validation, and Testing
Load and save checkpoints
Fast training & optimizations
Scripts
Backbones and Datasets registration
Reproducibility

Contents

Models
Toggle navigation of Models
Datasets
Toggle navigation of Datasets
- How to create a new dataset
- Custom schedulers
Backbone
Utils

How to ...

Run Mammoth
Upgrade to the new Mammoth!
Run STAR-Prompt

Awesome works with Mammoth

Our papers
Works that use Mammoth

Code reference

models
Toggle navigation of models
- Coda Prompt Utils
  Toggle navigation of Coda Prompt Utils
  - model
  - vit
- Deprecated
- Dualprompt Utils
  Toggle navigation of Dualprompt Utils
- L2p Utils
  Toggle navigation of L2p Utils
- Lora Prototype Utils
  Toggle navigation of Lora Prototype Utils
  - tuners
    Toggle navigation of tuners
    - full_tuner
    - ia3_tuner
    - lora_tuner
    - utils
  - fisher
  - generative_replay
  - lora_prompt
  - lora_vit
  - utils
- Moe Adapters Utils
  Toggle navigation of Moe Adapters Utils
  - adapter
  - clip
  - model
  - tokenizer
- Ranpac Utils
  Toggle navigation of Ranpac Utils
  - inc_net
  - ranpac
  - toolkit
  - vit
- Slca Utils
  Toggle navigation of Slca Utils
  - convs
    Toggle navigation of convs
    - cifar_resnet
    - linears
  - base
  - inc_net
  - slca
  - toolkit
- Star Utils
  Toggle navigation of Star Utils
  - star_perturber
- Tak Utils
  Toggle navigation of Tak Utils
  - backbone
  - fisher_diag
  - fisher_kfac
  - hooks
  - merging
  - templates
  - ties_merging
  - utils
- Twf Utils
  Toggle navigation of Twf Utils
  - afd
  - utils
- Utils
  Toggle navigation of Utils
- Zscl Utils
  Toggle navigation of Zscl Utils
  - clip
    Toggle navigation of clip
    - clip
    - model
    - tokenizer
  - cc
  - gather_cc
- agem
- agem_r
- attriclip
- bic
- ccic
- cgil
- clip
- cnll
- coda_prompt
- dap
- der
- derpp
- derpp_casper
- derpp_cscct
- derpp_lider
- derpp_star
- dualprompt
- er
- er_ace
- er_ace_aer_abs
- er_ace_casper
- er_ace_cscct
- er_ace_lider
- er_ace_star
- er_ace_tricks
- er_star
- er_tricks
- ewc_on
- fdr
- first_stage_starprompt
- gdumb
- gdumb_lider
- gem
- gss
- hal
- icarl
- icarl_casper
- icarl_cscct
- icarl_lider
- idefics
- joint
- joint_gcl
- l2p
- llava
- lucir
- lwf
- lwf_mc
- lws
- mer
- moe_adapters
- pnn
- puridiver
- ranpac
- rpc
- second_order
- second_stage_starprompt
- sgd
- si
- slca
- spr
- starprompt
- tak
- twf
- xder
- xder_ce
- xder_rpc
- xder_rpc_casper
- xder_rpc_cscct
- xder_rpc_star
- zscl
datasets
Toggle navigation of datasets
- Bias Celeba Utils
  Toggle navigation of Bias Celeba Utils
- Deprecated
  Toggle navigation of Deprecated
  - old_mnist_360
- Imagenet R Utils
- Transforms
  Toggle navigation of Transforms
- Utils
  Toggle navigation of Utils
- mnist_360
- perm_mnist
- rot_mnist
- seq_8vision
- seq_cars196
- seq_celeba
- seq_chestx
- seq_cifar10
- seq_cifar100
- seq_cifar100_224
- seq_cifar100_224_rs
- seq_cifar10_224
- seq_cifar10_224_rs
- seq_cropdisease
- seq_cub200
- seq_cub200_rs
- seq_dtd
- seq_eurosat_rgb
- seq_gtrsrb
- seq_imagenet_r
- seq_isic
- seq_mit67
- seq_mnist
- seq_mnist_224
- seq_resisc45
- seq_sun397
- seq_svhn
- seq_tinyimagenet
- seq_tinyimagenet_r
backbone
Toggle navigation of backbone
- Utils
  Toggle navigation of Utils
- EfficientNet
- MNISTMLP
- MNISTMLP_PNN
- ResNet18_PNN
- ResNet32
- ResNetBlock
- ResNetBottleneck
- vit
utils
Toggle navigation of utils
- args
- augmentations
- autoaugment
- batch_norm
- best_args
- bias
- bmm
- buffer
- buffer_lws
- checkpoints
- conditional_bn
- conf
- distributed
- evaluate
- globals
- gss_buffer
- kornia_utils
- loggers
- magic
- main
- metrics
- mixup
- notebooks
- prompt_templates
- ring_buffer
- schedulers
- simclrloss
- spkdloss
- stats
- status
- training
- triplet

Toggle table of contents sidebar

tak utils#

Copyright © 2026, Pietro Buzzega, Matteo Boschini, Lorenzo Bonicelli, Aniello Panariello, Davide Abati, Angelo Porrello, Simone Calderara

Made with Sphinx and @pradyunsg's Furo