File size: 390 Bytes
3bf7d18 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 | MODEL:
NAME: DracoDenoiseAutoencoder
DEVICE: cuda
IMG_SIZE: 1024
PATCH_SIZE: 32
IN_CHANS: 1
VIT_SCALE: base
DYNAMIC_IMG_SIZE: True
DYNAMIC_IMG_PAD: True
DECODER_EMBED_DIM: 512
DECODER_DEPTH: 8
DECODER_NUM_HEADS: 16
DECODER_USE_NECK: True
DECODER_NECK_DIM: 256
USE_ABS_POS: true
USE_DECODER_NECK: True
WINDOW_SIZE: 28
DECODER_GLOBAL_ATTN_INDEXES: [3, 7]
|