TheEditor / TransUnet_Config.py
SS3M's picture
Upload 7 files
e2cc14b verified
raw
history blame contribute delete
141 Bytes
in_channels = 3
adp_channels = 32
out_channels = 1
trans_num_layers = 5
trans_num_attn_heads = 8
trans_ffw_channels = 512
dropout = 0.1