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