draco / draco.yaml
Felix-Xu's picture
denoise model update
3bf7d18
raw
history blame contribute delete
390 Bytes
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]