SEnSeIv2 / sensei-configs /senseiv2-small.yaml
aliFrancis's picture
Add extra sensei configs
6f63715
raw
history blame
379 Bytes
---
descriptors:
style: SEnSeIv2
N_embeddings: 32
final_size: 74
blocks:
-
type: FCL
layer_sizes:
- 128
- 32
skips: add
-
type: BAND_EMBEDDING
embedding_dims: 32 # number of output channels
head_layer_sizes:
- 64
skips_heads: false
normalize: true