Skip to Content
DocsRf3Train

Train — RF3

saffron train \ activations_path=outputs/collect/vfdb_rf3/activations/activations.h5 \ hook_name=block12 \ steps=20000 \ use_wandb=true

Checkpoint written to outputs/train/matryoshka_batch_top_k/block12/<timestamp>/final.pt.

ParameterDefaultNotes
hook_nameOne of the hooks declared in the collect step
steps20000
batch_size4096
architecturematryoshka_batch_top_k
devicecuda:0
Last updated on